!5758 fix arg or hub in resnet

Merge pull request !5758 from zhouyaqiang0/master
pull/5758/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit e03e087079

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