diff --git a/PyTorch/contrib/cv/classification/MobileNet/test/train_full_1p.sh b/PyTorch/contrib/cv/classification/MobileNet/test/train_full_1p.sh
index 55b11273c94594277be939c7296414ba46b0f582..ff208ef246533b617a5339467dd0531b834ada0b 100755
--- a/PyTorch/contrib/cv/classification/MobileNet/test/train_full_1p.sh
+++ b/PyTorch/contrib/cv/classification/MobileNet/test/train_full_1p.sh
@@ -85,7 +85,7 @@ python3.7 mobilenet.py \
         --epochs=${train_epochs} \
         -j $(($(nproc))) \
         --lr=${learning_rate} \
-        --device_id={ASCEND_DEVICE_ID}  ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 &
+        --device_id=${ASCEND_DEVICE_ID}  ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 &
 
 wait
 
diff --git a/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_1p.sh b/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_1p.sh
index b790357f847dcec3d97783c56b4bf48225d5ea51..5ab530c3c039d5b831c167d50607e70d0abd7675 100755
--- a/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_1p.sh
+++ b/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_1p.sh
@@ -12,7 +12,7 @@ export RANK_SIZE=1
 data_path=""
 
 # 训练epoch
-train_epochs=1
+train_epochs=2
 # 指定训练所使用的npu device卡id
 device_id=0
 # 学习率
diff --git a/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_8p.sh b/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_8p.sh
index 0d8120807b069ced2135f8becf9a03b4e570d04b..373b7a034dc2efdbe38cb482742adb748823deab 100755
--- a/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_8p.sh
+++ b/PyTorch/contrib/cv/classification/MobileNet/test/train_performance_8p.sh
@@ -12,7 +12,7 @@ export RANK_SIZE=8
 data_path=""
 
 # 训练epoch
-train_epochs=1
+train_epochs=2
 # 学习率
 learning_rate=0.8
 
diff --git a/PyTorch/contrib/cv/classification/WideResNet101_2_for_Pytorch/test/train_performance_1p.sh b/PyTorch/contrib/cv/classification/WideResNet101_2_for_Pytorch/test/train_performance_1p.sh
index a05d07a3666e33dbcc6a60365bcca9ead50aba6f..0d05811ca367d24695c0ad78f8809ae66fdafa25 100644
--- a/PyTorch/contrib/cv/classification/WideResNet101_2_for_Pytorch/test/train_performance_1p.sh
+++ b/PyTorch/contrib/cv/classification/WideResNet101_2_for_Pytorch/test/train_performance_1p.sh
@@ -116,6 +116,7 @@ python3.7 -u ./main_npu_1p.py \
     --opt-level='O2' \
     --device='npu' \
     --world-size=1 \
+    --npu=${ASCEND_DEVICE_ID} \
     --batch-size=${batch_size} > ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 &
 
 wait