From a3f61800d8c37df52940194d4f7f1f29048cbd77 Mon Sep 17 00:00:00 2001 From: Oleg Ovcharenko Date: Tue, 12 Dec 2023 00:34:52 +0400 Subject: [PATCH] fix missing classes argument to ResNet --- MxNet/Classification/RN50v1.5/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/MxNet/Classification/RN50v1.5/models.py b/MxNet/Classification/RN50v1.5/models.py index 43b5ae099..5de06b99f 100644 --- a/MxNet/Classification/RN50v1.5/models.py +++ b/MxNet/Classification/RN50v1.5/models.py @@ -467,7 +467,7 @@ def create_resnet(builder, version, num_layers=50, resnext=False, classes=1000): block_class, layers, channels = resnet_spec[num_layers] assert not resnext or num_layers >= 50, \ "Cannot create resnext with less then 50 layers" - net = ResNet(builder, block_class, layers, channels, version=version, + net = ResNet(builder, block_class, layers, channels, classes, version=version, resnext_groups=args.num_groups if resnext else None) return net