From 5c9e62e352900b7f12c3c75f4c098ab3469b0375 Mon Sep 17 00:00:00 2001 From: andy0001 Date: Mon, 24 Apr 2023 11:38:33 +0800 Subject: [PATCH] Add FP32 --- .../MiniGo_ID0629_for_TensorFlow/test/train_full_1p.sh | 2 +- .../MiniGo_ID0629_for_TensorFlow/test/train_performance_1p.sh | 2 +- .../MiniGo_ID0629_for_TensorFlow/test/train_performance_8p.sh | 4 ++++ 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_full_1p.sh b/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_full_1p.sh index 9fcd87a82..0429cf38d 100644 --- a/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_full_1p.sh +++ b/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_full_1p.sh @@ -51,7 +51,7 @@ if [[ $data_path == "" ]];then fi if [[ $precision_mode == "must_keep_origin_dtype" ]];then - sed -i "s|allow_mix_precision|must_keep_origin_dtype|g" dual_net.py + sed -i "s|allow_mix_precision|must_keep_origin_dtype|g" $cur_path/dual_net.py fi diff --git a/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_performance_1p.sh b/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_performance_1p.sh index bdc87180a..35b336fdb 100644 --- a/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_performance_1p.sh +++ b/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_performance_1p.sh @@ -51,7 +51,7 @@ if [[ $data_path == "" ]];then fi if [[ $precision_mode == "must_keep_origin_dtype" ]];then - sed -i "s|allow_mix_precision|must_keep_origin_dtype|g" dual_net.py + sed -i "s|allow_mix_precision|must_keep_origin_dtype|g" $cur_path/dual_net.py fi ##############执行训练########## diff --git a/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_performance_8p.sh b/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_performance_8p.sh index 764068710..121618b0f 100644 --- a/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_performance_8p.sh +++ b/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_performance_8p.sh @@ -54,6 +54,10 @@ if [[ $data_path == "" ]];then exit 1 fi +if [[ $precision_mode == "must_keep_origin_dtype" ]];then + sed -i "s|allow_mix_precision|must_keep_origin_dtype|g" $cur_path/dual_net.py +fi + ##############执行训练########## cd $cur_path -- Gitee