From 71dc2dd485977200648ebaf802a14b8575e45466 Mon Sep 17 00:00:00 2001 From: YueJiang <1206052386@qq.com> Date: Thu, 10 Aug 2023 08:53:28 +0000 Subject: [PATCH] update test/train_performance_8p.sh. Signed-off-by: YueJiang <1206052386@qq.com> --- .../test/train_performance_8p.sh | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) 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 12bb9abc5..1352e2aa5 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 @@ -129,8 +129,11 @@ BatchSize=${batch_size} #设备类型,自动获取 DeviceType=`uname -m` #用例名称,自动获取 -CaseName=${Network}_bs${BatchSize}_${RankSize}'p'_'perf' - +if [[ $precision_mode == "must_keep_origin_dtype" ]];then + CaseName=${Network}_bs${BatchSize}_${RANK_SIZE}'p'_'fp32'_'perf' +else + CaseName=${Network}_bs${BatchSize}_${RANK_SIZE}'p'_'perf' +fi #获取性能 TrainingTime=`grep "tensorflow:global_step/sec" $cur_path/test/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log|awk 'END {print $2}'` wait -- Gitee