From 4ed03321c2b4ada3c1268617b1d3e4cf5cdf7b7d Mon Sep 17 00:00:00 2001 From: huchunmei Date: Tue, 8 Jun 2021 10:41:31 +0800 Subject: [PATCH] clould --- model_zoo/official/cv/alexnet/eval.py | 3 ++- tests/st/model_zoo_tests/maskrcnn/test_maskrcnn.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/model_zoo/official/cv/alexnet/eval.py b/model_zoo/official/cv/alexnet/eval.py index f5cef2d13ca..c84d7c93192 100644 --- a/model_zoo/official/cv/alexnet/eval.py +++ b/model_zoo/official/cv/alexnet/eval.py @@ -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) diff --git a/tests/st/model_zoo_tests/maskrcnn/test_maskrcnn.py b/tests/st/model_zoo_tests/maskrcnn/test_maskrcnn.py index 33a75dbe485..ab2d55c1e68 100644 --- a/tests/st/model_zoo_tests/maskrcnn/test_maskrcnn.py +++ b/tests/st/model_zoo_tests/maskrcnn/test_maskrcnn.py @@ -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)