sbjkx
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6637 modify resnet50_adv_pruning for hub loading 

Merge pull request !6637 from hzf/master
This commit is contained in:
mindspore-ci-bot 2020-09-21 19:23:08 +08:00 committed by Gitee
parent 5b3fcf810f 888a5412a5
commit a57c06f021
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
model_zoo/research/cv/resnet50_adv_pruning/mindpsore_hub_conf.py
View File

@ -14,9 +14,21 @@
# ============================================================================
"""hub config."""
from src.resnet_imgnet import resnet50
from mindspore import Tensor
import numpy as np
def create_network(name, *args, **kwargs):
if name == 'resnet-0.65x':
return resnet50(*args, **kwargs)
def get_index(filename):
index = []
with open(filename) as fr:
for line in fr:
ind = Tensor((np.array(line.strip('\n').split(' ')[:-1])).astype(np.int32).reshape(-1, 1))
index.append(ind)
return index
def create_network(name, rate=0.65, index_filename='index.txt', **kwargs):
index = get_index(index_filename)
if name == 'resnet50-0.65x':
return resnet50(rate=rate, index=index, **kwargs)
raise NotImplementedError(f"{name} is not implemented in the repo")