Skip to content

Commit 37bd11d

Browse files
szagoruykofmassa
authored andcommitted
Rename _resnet arguments (#910)
1 parent 9077164 commit 37bd11d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchvision/models/resnet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -204,8 +204,8 @@ def forward(self, x):
204204
return x
205205

206206

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

0 commit comments

Comments
 (0)