Browse Source

fix arg or hub in resnet

tags/v1.0.0
zhouyaqiang 5 years ago
parent
commit
5ad4cfe6c3
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/cv/resnet/mindspore_hub_conf.py

+ 1
- 1
model_zoo/official/cv/resnet/mindspore_hub_conf.py View File

@@ -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':


Loading…
Cancel
Save