Browse Source

!5758 fix arg or hub in resnet

Merge pull request !5758 from zhouyaqiang0/master
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
e03e087079
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