yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!1656 fix bug for mobilenet in model_zoo 

Merge pull request !1656 from SanjayChan/r0.3
This commit is contained in:
mindspore-ci-bot 2020-05-29 16:31:27 +08:00 committed by Gitee
parent 3ae2f8d12c e658eb7f24
commit 0e4574af6b
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
example/mobilenetv2_imagenet/eval.py
View File

@ -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')

2
example/mobilenetv3_imagenet/eval.py
View File

@ -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')