(self, x_rref)
| 132 | self.fc = nn.Linear(512 * self._block.expansion, num_classes).to(self.device) |
| 133 | |
| 134 | def forward(self, x_rref): |
| 135 | x = x_rref.to_here().to(self.device) |
| 136 | with self._lock: |
| 137 | out = self.fc(torch.flatten(self.seq(x), 1)) |
| 138 | return out.cpu() |
| 139 | |
| 140 | |
| 141 | class DistResNet50(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected