mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!16096 fix ssd and mobilenet_v1 issue 

From: @jiangzg001
Reviewed-by: @wuxuejian,@c_34
Signed-off-by: @wuxuejian
This commit is contained in:
mindspore-ci-bot 2021-05-10 11:32:12 +08:00 committed by Gitee
parent 889b9aab48 591030e0f9
commit dbce7882dd
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model_zoo/official/cv/mobilenetv1/README.md
View File

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

3
model_zoo/official/cv/ssd/src/ssd.py
View File

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