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.