!5758 fix arg or hub in resnet

Merge pull request !5758 from zhouyaqiang0/master
2020-09-17 19:48:36 +08:00 · 2020-09-17 19:48:36 +08:00 · e03e087079
parent c3656e8d7b 5ad4cfe6c3
commit e03e087079
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':