Skip to content

Commit a3f6180

Browse files
committed
fix missing classes argument to ResNet
1 parent 9dd9fcb commit a3f6180

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

MxNet/Classification/RN50v1.5/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -467,7 +467,7 @@ def create_resnet(builder, version, num_layers=50, resnext=False, classes=1000):
467467
block_class, layers, channels = resnet_spec[num_layers]
468468
assert not resnext or num_layers >= 50, \
469469
"Cannot create resnext with less then 50 layers"
470-
net = ResNet(builder, block_class, layers, channels, version=version,
470+
net = ResNet(builder, block_class, layers, channels, classes, version=version,
471471
resnext_groups=args.num_groups if resnext else None)
472472
return net
473473

0 commit comments

Comments
 (0)