From e37e4afc006e25b9a1cd006b6ae331edb4370822 Mon Sep 17 00:00:00 2001
From: limingxing517 <limingxing517@163.com>
Date: Wed, 1 Feb 2023 03:53:59 +0000
Subject: [PATCH] fp32
Signed-off-by: limingxing517 <limingxing517@163.com>
---
.../test/train_performance_1p.sh | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_1p.sh b/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_1p.sh
index a09fdda8a..2234ff78a 100644
--- a/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_1p.sh
+++ b/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_1p.sh
@@ -28,7 +28,7 @@ display_step=10
#维持参数，以下不需要修改
over_dump=False
-
+precision_mode="allow_mix_precision"
# 帮助信息，不需要修改
if [[ $1 == --help || $1 == -h ]];then
echo"usage:./train_performance_1p.sh <args>"
@@ -97,7 +97,11 @@ cp configs/config.py configs/config.py.run
#训练执行
start=$(date +%s)
-nohup python3 train.py --data_path=$data_path --ckpt_path=$cur_path/output/$ASCEND_DEVICE_ID/ckpt --train_size =$train_size --display_step=$display_step > $cur_path/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log 2>&1 &
+nohup python3 train.py --data_path=$data_path \
+ --ckpt_path=$cur_path/output/$ASCEND_DEVICE_ID/ckpt \
+ --train_size=$train_size \
+ --precision_mode=$precision_mode \
+ --display_step=$display_step > $cur_path/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log 2>&1 &
wait
end=$(date +%s)
e2e_time=$(( $end - $start ))
--
Gitee