p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!17957 maskrcnn repair 

From: @huchunmei
Reviewed-by: @wuxuejian,@c_34
Signed-off-by: @wuxuejian,@c_34
This commit is contained in:
mindspore-ci-bot 2021-06-08 14:43:46 +08:00 committed by Gitee
parent 8cd5eb68b3 4ed03321c2
commit f86d707126
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
model_zoo/official/cv/alexnet/eval.py
View File

@ -52,7 +52,8 @@ def eval_alexnet():
network = AlexNet(config.num_classes, phase='test') network = AlexNet(config.num_classes, phase='test')
loss = nn.SoftmaxCrossEntropyWithLogits(sparse=True, reduction="mean") loss = nn.SoftmaxCrossEntropyWithLogits(sparse=True, reduction="mean")
opt = nn.Momentum(network.trainable_params(), config.learning_rate, config.momentum) opt = nn.Momentum(network.trainable_params(), config.learning_rate, config.momentum)
ds_eval = create_dataset_cifar10(config, config.data_path, config.batch_size, target=config.device_target) ds_eval = create_dataset_cifar10(config, config.data_path, config.batch_size, status="test", \
target=config.device_target)
param_dict = load_checkpoint(config.ckpt_path) param_dict = load_checkpoint(config.ckpt_path)
print("load checkpoint from [{}].".format(config.ckpt_path)) print("load checkpoint from [{}].".format(config.ckpt_path))
load_param_into_net(network, param_dict) load_param_into_net(network, param_dict)

2
tests/st/model_zoo_tests/maskrcnn/test_maskrcnn.py
View File

@ -28,7 +28,7 @@ def test_maskrcnn_export():
export maskrcnn air. export maskrcnn air.
""" """
old_list = ["(config=config)", "(net, param_dict_new)"] old_list = ["(config=config)", "(net, param_dict_new)"]
new_list = ["(config=config\\n) '''", "(net, param_dict_new)\\n '''"] new_list = ["(config=config)\\n '''", "(net, param_dict_new)\\n '''"]
cur_path = os.getcwd() cur_path = os.getcwd()
model_path = "{}/../../../../model_zoo/official/cv".format(cur_path) model_path = "{}/../../../../model_zoo/official/cv".format(cur_path)