diff --git a/model_zoo/official/cv/resnet/mindspore_hub_conf.py b/model_zoo/official/cv/resnet/mindspore_hub_conf.py index 633905fe60..e9ffe87012 100644 --- 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':