From 2653483b52bfe53b4ffeb6d6cbdf0a21556a31d8 Mon Sep 17 00:00:00 2001
From: hanxiaoxuan <hanxiaoxuan1@huawei.com>
Date: Wed, 4 Jan 2023 16:43:56 +0800
Subject: [PATCH] add compile time 0104

---
 .../DIEN_ID3065_for_TensorFlow/script/train.py              | 2 +-
 .../test/train_RT2_performance_1p.sh                        | 6 +++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/TensorFlow/built-in/recommendation/DIEN_ID3065_for_TensorFlow/script/train.py b/TensorFlow/built-in/recommendation/DIEN_ID3065_for_TensorFlow/script/train.py
index 8f99a5799..d866126f1 100644
--- a/TensorFlow/built-in/recommendation/DIEN_ID3065_for_TensorFlow/script/train.py
+++ b/TensorFlow/built-in/recommendation/DIEN_ID3065_for_TensorFlow/script/train.py
@@ -233,7 +233,7 @@ def train(
                 aux_loss_sum += aux_loss
                 iter += 1
                 sys.stdout.flush()
-                if (iter % test_iter) == 0:
+                if (iter % test_iter) == 0 or iter <= 5:
                     avg_examples_per_second = batch_size/(end_time - start_time)
                     print("avg_examples_per_second: ", avg_examples_per_second)
                     print('iter: %d ----> train_loss: %.4f ---- train_accuracy: %.4f ---- train_aux_loss: %.4f ---- perf: %.4f' % \
diff --git a/TensorFlow/built-in/recommendation/DIEN_ID3065_for_TensorFlow/test/train_RT2_performance_1p.sh b/TensorFlow/built-in/recommendation/DIEN_ID3065_for_TensorFlow/test/train_RT2_performance_1p.sh
index a397bc8f7..e46adfa36 100644
--- a/TensorFlow/built-in/recommendation/DIEN_ID3065_for_TensorFlow/test/train_RT2_performance_1p.sh
+++ b/TensorFlow/built-in/recommendation/DIEN_ID3065_for_TensorFlow/test/train_RT2_performance_1p.sh
@@ -131,10 +131,13 @@ sed -i "s|break|pass|g" train.py
 #结果打印，不需要修改
 echo "------------------ Final result ------------------"
 # #输出性能FPS，需要模型审视修改
-FPS=`grep  avg_examples_per_second $cur_path/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log | awk -F ":" '{print $2}' |sed s/[[:space:]]//g | tail -n +1 | awk '{sum+=$1} END {print sum/NR}'`
+FPS=`grep  avg_examples_per_second $cur_path/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log | awk -F ":" '{print $2}' |sed s/[[:space:]]//g | tail -n +6 | awk '{sum+=$1} END {print sum/NR}'`
 #打印，不需要修改
 echo "Final Performance item/sec : $FPS"
 
+#输出CompileTime
+CompileTime=`grep "perf:" $cur_path/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log |head -n 1|awk '{print $14}'`
+
 #输出训练精度,需要模型审视修改
 train_accuracy=`grep "train_accuracy" $cur_path/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log | awk -F "train_accuracy:" 'END{print $2}' | awk -F ' ' '{print $1}'|sed s/[[:space:]]//g`
 #打印，不需要修改
@@ -170,4 +173,5 @@ echo "TrainAccuracy = ${train_accuracy}" >> $cur_path/output/$ASCEND_DEVICE_ID/$
 echo "TrainingTime = ${TrainingTime}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
 echo "ActualLoss = ${ActualLoss}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
 echo "E2ETrainingTime = ${e2e_time}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
+echo "CompileTime = ${CompileTime}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log
 
-- 
Gitee