diff --git a/torchvision/models/resnet.py b/torchvision/models/resnet.py index 287e35fc9db..0727c731aec 100644 --- a/torchvision/models/resnet.py +++ b/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, stride=1) + self.avgpool = nn.AvgPool2d(7)#, stride=1) self.fc = nn.Linear(512 * block.expansion, num_classes) for m in self.modules(): diff --git a/torchvision/transforms/transforms.py b/torchvision/transforms/transforms.py index bc64bef2a29..0af7f352d9b 100644 --- a/torchvision/transforms/transforms.py +++ b/torchvision/transforms/transforms.py @@ -341,8 +341,7 @@ def __init__(self, size, scale=(0.08, 1.0), ratio=(3. / 4., 4. / 3.), interpolat self.scale = scale self.ratio = ratio - @staticmethod - def get_params(img): + def get_params(self, img): """Get parameters for ``crop`` for a random sized crop. Args: