From 5ff5c36ee548b2b94f75d1ced40c7c15d9cb6b32 Mon Sep 17 00:00:00 2001 From: jiangzhenguang Date: Sat, 8 May 2021 14:15:53 +0800 Subject: [PATCH] fix issue --- model_zoo/official/cv/mobilenetv1/README.md | 3 ++- model_zoo/official/cv/ssd/src/ssd.py | 3 +-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/model_zoo/official/cv/mobilenetv1/README.md b/model_zoo/official/cv/mobilenetv1/README.md index 15b858359c..31142ea1f2 100644 --- a/model_zoo/official/cv/mobilenetv1/README.md +++ b/model_zoo/official/cv/mobilenetv1/README.md @@ -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] diff --git a/model_zoo/official/cv/ssd/src/ssd.py b/model_zoo/official/cv/ssd/src/ssd.py index f030f75dcb..d876a5b910 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