pretrained-models.pytorch icon indicating copy to clipboard operation
pretrained-models.pytorch copied to clipboard

Change last pooling layer in SENet

Open dattran2346 opened this issue 4 years ago • 0 comments

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.

dattran2346 avatar Sep 20 '19 08:09 dattran2346