(self, x: torch.Tensor)
| 37 | print(f"MyModel: {n_params / 1e6:.1f}M parameters") |
| 38 | |
| 39 | def forward(self, x: torch.Tensor) -> torch.Tensor: |
| 40 | x = self.features(x) |
| 41 | x = x.flatten(1) |
| 42 | x = self.classifier(x) |
| 43 | return x |
nothing calls this directly
no outgoing calls
no test coverage detected