(self, model_input)
| 451 | return outputs |
| 452 | |
| 453 | def forward(self, model_input): |
| 454 | |
| 455 | coords = model_input['coords'] |
| 456 | |
| 457 | if self.integrated_pe: |
| 458 | coords = [self.pe(coords, r) for r in model_input['radii']] |
| 459 | elif self.nl != 'sine' and not self.no_pe: |
| 460 | coords = self.pe(coords) |
| 461 | |
| 462 | output = self.net_forward(coords) |
| 463 | |
| 464 | return {'model_in': model_input, 'model_out': {'output': output}} |
| 465 | |
| 466 | |
| 467 | class CoordinateNet(nn.Module): |
nothing calls this directly
no test coverage detected