Browse Source

!1656 fix bug for mobilenet in model_zoo

Merge pull request !1656 from SanjayChan/r0.3
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
0e4574af6b
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      example/mobilenetv2_imagenet/eval.py
  2. +1
    -1
      example/mobilenetv3_imagenet/eval.py

+ 1
- 1
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')


+ 1
- 1
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')


Loading…
Cancel
Save