pretrained-models.pytorch
pretrained-models.pytorch copied to clipboard
Change last pooling layer in SENet
The last pooling layer of SENet should be change from self.avg_pool = nn.AvgPool2d(7, stride=1) to self.avg_pool = nn.AdaptiveAvgPool2d((1, 1)) so that the model can take different input size.