diff --git a/TensorFlow/built-in/cv/detection/SSD-VGG_ID1619_for_TensorFlow/test/train_performance_1p.sh b/TensorFlow/built-in/cv/detection/SSD-VGG_ID1619_for_TensorFlow/test/train_performance_1p.sh index 0004076aa3695ba342771864a8ae6c77b1649e99..e663985dcf3d365a89b4a94157f5137133bf2fb4 100644 --- a/TensorFlow/built-in/cv/detection/SSD-VGG_ID1619_for_TensorFlow/test/train_performance_1p.sh +++ b/TensorFlow/built-in/cv/detection/SSD-VGG_ID1619_for_TensorFlow/test/train_performance_1p.sh @@ -63,8 +63,7 @@ start=$(date +%s) nohup python3 train.py \ --epochs ${train_epochs} \ --data-dir $data_path/pascal-voc \ - --vgg-dir $data_path/vgg_graph \ - "${@:1}" > $cur_path/test/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log 2>&1 & + --vgg-dir $data_path/vgg_graph > $cur_path/test/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log 2>&1 & wait end=$(date +%s) diff --git a/TensorFlow/contrib/cv/deeplab-v3-plus/deeplabv3+_hw09124698/test/train_perf_1p.sh b/TensorFlow/contrib/cv/deeplab-v3-plus/deeplabv3+_hw09124698/test/train_perf_1p.sh index fa0e0fbdfa024a6bc3b6b61a9a18e2051267a103..c08ded6960345a4a31e4d8ab2a92c76e94d5132f 100644 --- a/TensorFlow/contrib/cv/deeplab-v3-plus/deeplabv3+_hw09124698/test/train_perf_1p.sh +++ b/TensorFlow/contrib/cv/deeplab-v3-plus/deeplabv3+_hw09124698/test/train_perf_1p.sh @@ -106,10 +106,12 @@ e2e_time=$(( $end_time - $start_time )) #结果打印,不需要修改 echo "------------------ Final result ------------------" #输出性能FPS,需要模型审视修改 -step_sec=`grep -a 'INFO:tensorflow:global_step/sec: ' ${cur_path}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log|awk 'END {print $2}'` + +i=`grep -a 'INFO:tensorflow:global_step/sec: ' ${cur_path}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log|awk 'END {print $7}'` +step_sec=${i##*(} #FPS=`grep "Final Performance images/sec" ${cur_path}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log|awk '{print $5}'` #FPS=`expr ${RANK_SIZE} \* ${FPS}` -FPS=`awk -v x="$batch_size" -v y="$step_sec" 'BEGIN{printf "%.2f\n",x*y}'` +FPS=`awk -v x="$batch_size" -v y="$step_sec" 'BEGIN{printf "%.2f\n",x/y}'` FPS=`awk -v x="$RANK_SIZE" -v y="$FPS" 'BEGIN{printf "%.2f\n",x*y}'` #打印,不需要修改 echo "Final Performance images/sec : $FPS"