From e658eb7f244eb267ead28bd54bf44da6c0ec5d1d Mon Sep 17 00:00:00 2001 From: chenzomi Date: Fri, 29 May 2020 11:26:50 +0800 Subject: [PATCH] bug fix --- example/mobilenetv2_imagenet/eval.py | 2 +- example/mobilenetv3_imagenet/eval.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/example/mobilenetv2_imagenet/eval.py b/example/mobilenetv2_imagenet/eval.py index 61e9c5ecd7f..d8e25ff93b9 100644 --- a/example/mobilenetv2_imagenet/eval.py +++ b/example/mobilenetv2_imagenet/eval.py @@ -22,9 +22,9 @@ from mindspore import nn from mindspore.train.model import Model from mindspore.train.serialization import load_checkpoint, load_param_into_net from mindspore.common import dtype as mstype +from mindspore.model_zoo.mobilenetV2 import mobilenet_v2 from src.dataset import create_dataset from src.config import config_ascend, config_gpu -from src.mobilenetV2 import mobilenet_v2 parser = argparse.ArgumentParser(description='Image classification') diff --git a/example/mobilenetv3_imagenet/eval.py b/example/mobilenetv3_imagenet/eval.py index e82ed496d3f..a8f5138b4f7 100644 --- a/example/mobilenetv3_imagenet/eval.py +++ b/example/mobilenetv3_imagenet/eval.py @@ -22,9 +22,9 @@ from mindspore import nn from mindspore.train.model import Model from mindspore.train.serialization import load_checkpoint, load_param_into_net from mindspore.common import dtype as mstype +from mindspore.model_zoo.mobilenetV3 import mobilenet_v3_large from src.dataset import create_dataset from src.config import config_ascend, config_gpu -from src.mobilenetV3 import mobilenet_v3_large parser = argparse.ArgumentParser(description='Image classification')