diff --git a/PyTorch/dev/cv/detection/SSD-ResNet34_ID0411_for_PyTorch/test/train_performance_16p.sh b/PyTorch/dev/cv/detection/SSD-ResNet34_ID0411_for_PyTorch/test/train_performance_16p.sh
index e7981d708560deee4c9138f9f865e1dc2d0d42da..16bb5fcea9c61d07220f8a6f2ea2f9bac103838f 100644
--- a/PyTorch/dev/cv/detection/SSD-ResNet34_ID0411_for_PyTorch/test/train_performance_16p.sh
+++ b/PyTorch/dev/cv/detection/SSD-ResNet34_ID0411_for_PyTorch/test/train_performance_16p.sh
@@ -114,7 +114,7 @@ fi
 wait
 
 
-
+export RankSize=16
 export HCCL_IF_IP=$fix_node_ip
 export MASTER_ADDR=$one_node_ip
 export MASTER_PORT=29688
@@ -200,7 +200,7 @@ echo "Final Performance images/sec : $FPS"
 #训练用例信息，不需要修改
 BatchSize=${batch_size}
 DeviceType=`uname -m`
-CaseName=${Network}_bs${BatchSize}_${RANK_SIZE}'p'_'perf'
+CaseName=${Network}_bs${BatchSize}_${RankSize}'p'_'perf'
 
 ##获取性能数据
 #吞吐量，不需要修改
@@ -216,7 +216,7 @@ ActualLoss=`awk 'END {print}' $cur_path/output/$ASCEND_DEVICE_ID/train_${CaseNam
 
 #关键信息打印到${CaseName}.log中，不需要修改
 echo "Network = ${Network}" > $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
-echo "RankSize = ${RANK_SIZE}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
+echo "RankSize = ${RankSize}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
 echo "BatchSize = ${BatchSize}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
 echo "DeviceType = ${DeviceType}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
 echo "CaseName = ${CaseName}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
diff --git a/PyTorch/dev/cv/image_classification/DeepSort_ID0654_for_PyTorch/test/train_performance_8p.sh b/PyTorch/dev/cv/image_classification/DeepSort_ID0654_for_PyTorch/test/train_performance_8p.sh
index 1caa51cfe889539996085e509bb8d388b90b885c..10ad32519d09c160debb986eafb590c1358ca018 100644
--- a/PyTorch/dev/cv/image_classification/DeepSort_ID0654_for_PyTorch/test/train_performance_8p.sh
+++ b/PyTorch/dev/cv/image_classification/DeepSort_ID0654_for_PyTorch/test/train_performance_8p.sh
@@ -98,6 +98,7 @@ else
 fi
 wait
 
+export RankSize=8
 #训练开始时间，不需要修改
 start_time=$(date +%s)
 
@@ -171,7 +172,7 @@ echo "E2E Training Duration sec : $e2e_time"
 #训练用例信息，不需要修改
 BatchSize=${batch_size}
 DeviceType=`uname -m`
-CaseName=${Network}_bs${BatchSize}_${RANK_SIZE}'p'_'perf'
+CaseName=${Network}_bs${BatchSize}_${RankSize}'p'_'perf'
 
 ##获取性能数据，不需要修改
 #吞吐量
@@ -188,7 +189,7 @@ ActualLoss=`awk 'END {print}' $cur_path/output/$ASCEND_DEVICE_ID/train_${CaseNam
 
 #关键信息打印到${CaseName}.log中，不需要修改
 echo "Network = ${Network}" > $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
-echo "RankSize = ${RANK_SIZE}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
+echo "RankSize = ${RankSize}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
 echo "BatchSize = ${BatchSize}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
 echo "DeviceType = ${DeviceType}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
 echo "CaseName = ${CaseName}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log