| 70 | |
| 71 | |
| 72 | class PreprocessingModel(nn.Module): |
| 73 | def __init__(self, resize, mean, std): |
| 74 | super(PreprocessingModel, self).__init__() |
| 75 | self.resize = transforms.Resize(resize) |
| 76 | self.normalize = transforms.Normalize(mean, std) |
| 77 | |
| 78 | def forward(self, x): |
| 79 | return self.normalize(self.resize(x)) |
| 80 | |
| 81 | |
| 82 | class EnsembleModel(torch.nn.Module): |
no outgoing calls
no test coverage detected