fix arg or hub in resnet

5 years ago · 5ad4cfe6c3
parent e540b88e3f
commit 5ad4cfe6c3
1 changed files with 1 additions and 1 deletions
--- a/model_zoo/official/cv/resnet/mindspore_hub_conf.py
+++ b/model_zoo/official/cv/resnet/mindspore_hub_conf.py
@ -15,7 +15,7 @@
 """hub config."""
 from src.resnet import resnet50, resnet101, se_resnet50

-def create_network(name, **kwargs):
+def create_network(name, *args, **kwargs):
    if name == 'resnet50':
        return resnet50(*args, **kwargs)
    if name == 'resnet101':