Rename `_resnet` arguments (#910)

37bd11df · Sergey Zagoruyko · Francisco Massa · 90771646 · 37bd11df
Commit 37bd11df authored May 17, 2019 by Sergey Zagoruyko Committed by Francisco Massa May 17, 2019
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torchvision/models/resnet.py torchvision/models/resnet.py +2 -2

No files found.
--- a/torchvision/models/resnet.py
+++ b/torchvision/models/resnet.py
@@ -204,8 +204,8 @@ class ResNet(nn.Module):
        return x


-def _resnet(arch, inplanes, planes, pretrained, progress, **kwargs):
-    model = ResNet(inplanes, planes, **kwargs)
+def _resnet(arch, block, layers, pretrained, progress, **kwargs):
+    model = ResNet(block, layers, **kwargs)
    if pretrained:
        state_dict = load_state_dict_from_url(model_urls[arch],
                                              progress=progress)