diff --git a/PyTorch/contrib/cv/detection/SOLOv2/test/train_eval_1p.sh b/PyTorch/contrib/cv/detection/SOLOv2/test/train_eval_1p.sh index 8132b286e3e8c50a53cde219bb12f297c1288f03..7f5f6c5617d05e5a37a66dbe838fe903f056c986 100644 --- a/PyTorch/contrib/cv/detection/SOLOv2/test/train_eval_1p.sh +++ b/PyTorch/contrib/cv/detection/SOLOv2/test/train_eval_1p.sh @@ -70,7 +70,7 @@ end_time=$(date +%s) e2e_time=$(( $end_time - $start_time )) # 输出训练精度,需要模型审视修改 # eval.log | awk -F ',' '{print $1}' | awk '{print $2}' | awk ' END {print}' -train_accuracy=`grep "bbox_mAP" ${cur_path}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log | awk -F ',' '{print $1}' | awk '{print $2}' | awk ' END {print}'` +train_accuracy=`grep -a 'maxDets' $cur_path/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log|awk -F " " '{print $13}'|head -n 1` # 打印,不需要修改 echo "Final Train Accuracy : ${train_accuracy}" echo "E2E Training Duration sec : $e2e_time" diff --git a/PyTorch/contrib/cv/detection/SOLOv2/test/train_full_1p.sh b/PyTorch/contrib/cv/detection/SOLOv2/test/train_full_1p.sh index b4be5448561ecc86dbfde2a078eba4c9464dc3f0..52f8cc26803e0024034d6a98f8ea197ef880a4bc 100644 --- a/PyTorch/contrib/cv/detection/SOLOv2/test/train_full_1p.sh +++ b/PyTorch/contrib/cv/detection/SOLOv2/test/train_full_1p.sh @@ -86,7 +86,7 @@ FPS=`grep -a 'FPS' $cur_path/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_I echo "Final Performance images/sec : $FPS" #输出训练精度,需要模型审视修改 -train_accuracy=`grep -a 'bbox_mAP' $cur_path/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log|awk -F "bbox_mAP: " '{print $NF}'|awk -F "," '{print $1}'|awk 'NR==1{max=$1;next}{max=max>$1?max:$1}END{print max}'` +train_accuracy=`grep -a 'maxDets' $cur_path/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log|awk -F " " '{print $13}'|head -n 1` #打印,不需要修改 echo "Final Train Accuracy : ${train_accuracy}" echo "E2E Training Duration sec : $e2e_time" diff --git a/PyTorch/contrib/cv/detection/SOLOv2/test/train_full_8p.sh b/PyTorch/contrib/cv/detection/SOLOv2/test/train_full_8p.sh index 00f40316f47639de8d8d34704b4682d68ab3eb36..bf045521f0597f74fe0800c560b0669b3bb41b33 100644 --- a/PyTorch/contrib/cv/detection/SOLOv2/test/train_full_8p.sh +++ b/PyTorch/contrib/cv/detection/SOLOv2/test/train_full_8p.sh @@ -112,7 +112,7 @@ FPS=`grep -a 'FPS' $cur_path/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_I echo "Final Performance images/sec : $FPS" #输出训练精度,需要模型审视修改 -train_accuracy=`grep -a 'bbox_mAP' $cur_path/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log|awk -F "bbox_mAP: " '{print $NF}'|awk -F "," '{print $1}'|awk 'NR==1{max=$1;next}{max=max>$1?max:$1}END{print max}'` +train_accuracy=`grep -a 'maxDets' $cur_path/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log|awk -F " " '{print $13}'|head -n 1` #打印,不需要修改 echo "Final Train Accuracy : ${train_accuracy}" echo "E2E Training Duration sec : $e2e_time"