Browse Source

fix issue

pull/16101/head
jiangzhenguang 4 years ago
parent
commit
5ff5c36ee5
2 changed files with 3 additions and 3 deletions
  1. +2
    -1
      model_zoo/official/cv/mobilenetv1/README.md
  2. +1
    -2
      model_zoo/official/cv/ssd/src/ssd.py

+ 2
- 1
model_zoo/official/cv/mobilenetv1/README.md View File

@@ -139,7 +139,8 @@ You can start training using python or shell scripts.If the train method is trai
```shell
# eval example
python:
Ascend: python eval.py --dataset [cifar10|imagenet2012] --dataset_path [VAL_DATASET_PATH] --pretrain_ckpt [CHECKPOINT_PATH]
Ascend: python eval.py --dataset [cifar10|imagenet2012] --dataset_path [VAL_DATASET_PATH] --checkpoint_path [CHECKPOINT_PATH]
CPU: python eval.py --dataset [cifar10|imagenet2012] --dataset_path [VAL_DATASET_PATH] --checkpoint_path [CHECKPOINT_PATH] --device_target CPU

shell:
Ascend: sh run_eval.sh [cifar10|imagenet2012] [DATASET_PATH] [CHECKPOINT_PATH]


+ 1
- 2
model_zoo/official/cv/ssd/src/ssd.py View File

@@ -134,7 +134,6 @@ class InvertedResidual(nn.Cell):
_bn(oup),
])
self.conv = nn.SequentialCell(layers)
self.add = P.Add()
self.cast = P.Cast()
self.last_relu = last_relu
self.relu = nn.ReLU6()
@@ -143,7 +142,7 @@ class InvertedResidual(nn.Cell):
identity = x
x = self.conv(x)
if self.use_res_connect:
x = self.add(identity, x)
x = identity + x
if self.last_relu:
x = self.relu(x)
return x


Loading…
Cancel
Save