From 5ad4cfe6c37b75128ccc61db40c3a10c27ae6254 Mon Sep 17 00:00:00 2001 From: zhouyaqiang Date: Fri, 4 Sep 2020 14:16:13 +0800 Subject: [PATCH] fix arg or hub in resnet --- model_zoo/official/cv/resnet/mindspore_hub_conf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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':