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

clould

This commit is contained in:
huchunmei 2021-06-08 10:41:31 +08:00
parent 7e5d68991a
commit 4ed03321c2
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')
loss = nn.SoftmaxCrossEntropyWithLogits(sparse=True, reduction="mean")
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)
print("load checkpoint from [{}].".format(config.ckpt_path))
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.
"""
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()
model_path = "{}/../../../../model_zoo/official/cv".format(cur_path)