From 3393a47c04e7153cf5b0622246e0f77020b8d8eb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=AB=98=E5=85=B4=E6=88=90?= <1358493914@qq.com>
Date: Mon, 25 Apr 2022 03:14:42 +0000
Subject: [PATCH] update
 PyTorch/dev/cv/image_classification/SG2IM_ID0786_for_PyTorch/test/train_performance_1p.sh.

---
 .../SG2IM_ID0786_for_PyTorch/test/train_performance_1p.sh | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/PyTorch/dev/cv/image_classification/SG2IM_ID0786_for_PyTorch/test/train_performance_1p.sh b/PyTorch/dev/cv/image_classification/SG2IM_ID0786_for_PyTorch/test/train_performance_1p.sh
index 676246921c..e5171ef512 100644
--- a/PyTorch/dev/cv/image_classification/SG2IM_ID0786_for_PyTorch/test/train_performance_1p.sh
+++ b/PyTorch/dev/cv/image_classification/SG2IM_ID0786_for_PyTorch/test/train_performance_1p.sh
@@ -170,6 +170,7 @@ grep "total_loss" $cur_path/test/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_I
 
 #最后一个迭代loss值，不需要修改
 ActualLoss=`awk 'END {print}' $cur_path/test/output/$ASCEND_DEVICE_ID/train_${CaseName}_loss.txt`
+Status=`grep "${error_msg}" $cur_path/test/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log | wc -l`
 
 #关键信息打印到${CaseName}.log中，不需要修改
 echo "Network = ${Network}" > $cur_path/test/output/$ASCEND_DEVICE_ID/${CaseName}.log
@@ -177,7 +178,8 @@ echo "RankSize = ${RANK_SIZE}" >> $cur_path/test/output/$ASCEND_DEVICE_ID/${Case
 echo "BatchSize = ${BatchSize}" >> $cur_path/test/output/$ASCEND_DEVICE_ID/${CaseName}.log
 echo "DeviceType = ${DeviceType}" >> $cur_path/test/output/$ASCEND_DEVICE_ID/${CaseName}.log
 echo "CaseName = ${CaseName}" >> $cur_path/test/output/$ASCEND_DEVICE_ID/${CaseName}.log
-echo "ActualFPS = ${ActualFPS}" >> $cur_path/test/output/$ASCEND_DEVICE_ID/${CaseName}.log
-echo "TrainingTime = ${TrainingTime}" >> $cur_path/test/output/$ASCEND_DEVICE_ID/${CaseName}.log
-echo "ActualLoss = ${ActualLoss}" >> $cur_path/test/output/$ASCEND_DEVICE_ID/${CaseName}.log
+#echo "ActualFPS = ${ActualFPS}" >> $cur_path/test/output/$ASCEND_DEVICE_ID/${CaseName}.log
+#echo "TrainingTime = ${TrainingTime}" >> $cur_path/test/output/$ASCEND_DEVICE_ID/${CaseName}.log
+#echo "ActualLoss = ${ActualLoss}" >> $cur_path/test/output/$ASCEND_DEVICE_ID/${CaseName}.log
 echo "E2ETrainingTime = ${e2e_time}" >> $cur_path/test/output/$ASCEND_DEVICE_ID/${CaseName}.log
+echo "Status = ${Status}" >> $cur_path/test/output/$ASCEND_DEVICE_ID/${CaseName}.log
-- 
Gitee