(self, x: Tensor)
| 95 | |
| 96 | # Support torch.script function |
| 97 | def _forward_impl(self, x: Tensor) -> Tensor: |
| 98 | x = self.features(x) |
| 99 | x = self.avgpool(x) |
| 100 | x = torch.flatten(x, 1) |
| 101 | x = self.classifier(x) |
| 102 | |
| 103 | return x |
| 104 | |
| 105 | |
| 106 | class SRResNet(nn.Module): |