diff --git a/model_zoo/official/cv/mobilenetv1/README.md b/model_zoo/official/cv/mobilenetv1/README.md index c800518f7ef..75c6faa9102 100644 --- a/model_zoo/official/cv/mobilenetv1/README.md +++ b/model_zoo/official/cv/mobilenetv1/README.md @@ -143,8 +143,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] - CPU: python eval.py --dataset [cifar10|imagenet2012] --dataset_path [VAL_DATASET_PATH] --pretrain_ckpt [CHECKPOINT_PATH] --device_target CPU + 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] diff --git a/model_zoo/official/cv/ssd/src/ssd.py b/model_zoo/official/cv/ssd/src/ssd.py index f030f75dcb3..d876a5b910f 100644 --- a/model_zoo/official/cv/ssd/src/ssd.py +++ b/model_zoo/official/cv/ssd/src/ssd.py @@ -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