change resnet Avgpool2d to Adaptive_AvgPool_2d

Author: XavierLinNow

torchvision/models/resnet.py

@@ -107,7 +107,7 @@ def __init__(self, block, layers, num_classes=1000):
  self.layer2 = self._make_layer(block, 128, layers[1], stride=2)
  self.layer3 = self._make_layer(block, 256, layers[2], stride=2)
  self.layer4 = self._make_layer(block, 512, layers[3], stride=2)
- self.avgpool = nn.AvgPool2d(7)
+ self.avgpool = nn.AdaptiveAvgPool2d(1)
  self.fc = nn.Linear(512 * block.expansion, num_classes)
 
  for m in self.modules():