From 78509a943852d8b02ba6d6904735f8b00b7c0ecf Mon Sep 17 00:00:00 2001 From: jiangxing001 <1484736931@qq.com> Date: Fri, 10 Mar 2023 02:26:03 +0000 Subject: [PATCH] update TensorFlow/built-in/cv/detection/NIMA_ID0158_for_TensorFlow/test/train_performance_1p.sh. Signed-off-by: jiangxing001 <1484736931@qq.com> --- .../test/train_performance_1p.sh | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/TensorFlow/built-in/cv/detection/NIMA_ID0158_for_TensorFlow/test/train_performance_1p.sh b/TensorFlow/built-in/cv/detection/NIMA_ID0158_for_TensorFlow/test/train_performance_1p.sh index 56e070367..16f3a6057 100644 --- a/TensorFlow/built-in/cv/detection/NIMA_ID0158_for_TensorFlow/test/train_performance_1p.sh +++ b/TensorFlow/built-in/cv/detection/NIMA_ID0158_for_TensorFlow/test/train_performance_1p.sh @@ -91,8 +91,19 @@ e2e_time=$(( $end_time - $start_time )) #结果打印,不需要修改 echo "------------------ Final result ------------------" #输出性能FPS,需要模型审视修改 -train_time=`grep '1250/1250' $cur_path/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log|awk -F 'ms/step' '{print $1}'|awk '{print $NF}'` -FPS=`awk 'BEGIN{printf "%.2f\n", '$batch_size'* 1000/'$train_time'}'` +tp=`grep 'ms/step' $cur_path/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log | awk -F " " 'END{print$5}'` +echo $tp +if [ -z $tp ]; then + echo "IS NULL" + train_time=`grep '1250/1250' $cur_path/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log|awk -F 's/step' '{print $1}'|awk '{print $NF}'` + FPS=`awk 'BEGIN{printf "%.2f\n", '$batch_size'/'$train_time'}'` +else + echo "NOT NULL" + train_time=`grep '1250/1250' $cur_path/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log|awk -F 'ms/step' '{print $1}'|awk '{print $NF}'` + FPS=`awk 'BEGIN{printf "%.2f\n", '$batch_size'* 1000/'$train_time'}'` +fi + + #打印,不需要修改 echo "Final Performance images/sec : $FPS" -- Gitee