From af0a0ec42c3680067cc7412613bfde1edc6e896f Mon Sep 17 00:00:00 2001
From: YueJiang <1206052386@qq.com>
Date: Mon, 21 Aug 2023 10:30:38 +0000
Subject: [PATCH] update train_RT2_performance_8p.sh.

Signed-off-by: YueJiang <1206052386@qq.com>
---
 .../test/train_RT2_performance_8p.sh                           | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_RT2_performance_8p.sh b/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_RT2_performance_8p.sh
index 8b3c78b5d..392b36783 100644
--- a/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_RT2_performance_8p.sh
+++ b/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_RT2_performance_8p.sh
@@ -92,6 +92,7 @@ do
         mkdir -p $cur_path/test/output/$ASCEND_DEVICE_ID
     fi
     echo $ASCEND_DEVICE_ID
+    mkdir -p $cur_path/outputs/$ASCEND_DEVICE_ID/models/
     #(Step3)训练
     corenum=`cat /proc/cpuinfo |grep 'processor' |wc -l`
     let a=RANK_IDS*${corenum}/8
@@ -106,7 +107,7 @@ do
 		--steps_to_train=$train_steps \
 		--train_batch_size=$batch_size \
 		--work_dir=$cur_path/estimator_working_dir \
-		--export_path=$cur_path/outputs/models/000001-first_generation \
+		--export_path=$cur_path/outputs/$ASCEND_DEVICE_ID/models/000001-first_generation \
 		--dynamic_input=${dynamic_input} \
 		--jit_compile=False > $cur_path/test/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log 2>&1 &
 done
-- 
Gitee