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 8b3c78b5df45f61d92e60b7e11db56418677b32b..392b367832931410a54b2193be1a9d78e564283f 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 diff --git a/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_full_8p.sh b/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_full_8p.sh index 2ef53ea50fa09176ab6c4859631c1d5e8382725c..e996cf3d54296f1494c96385b686646f612beeb5 100644 --- a/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_full_8p.sh +++ b/TensorFlow/built-in/cv/image_classification/MiniGo_ID0629_for_TensorFlow/test/train_full_8p.sh @@ -88,6 +88,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_ID*${corenum}/8 @@ -96,7 +97,7 @@ do if [ "x${bind_core}" != x ];then bind_core="taskset -c $a-$c" fi - ${bind_core} python3 train.py --training_data_path=$data_path --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 --dynamic_input=${dynamic_input}> $cur_path/test/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log 2>&1 & + ${bind_core} python3 train.py --training_data_path=$data_path --steps_to_train=$train_steps --train_batch_size=$batch_size --work_dir=$cur_path/estimator_working_dir --export_path=$cur_path/outputs/$ASCEND_DEVICE_ID/models/000001-first_generation --dynamic_input=${dynamic_input}> $cur_path/test/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log 2>&1 & done wait 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 3d86d6562e771b580f489c8f0506e50242de7ce6..5e5a34855b4403dc96e90211b7be41228d47de10 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 @@ -98,6 +98,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 @@ -112,7 +113,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} > $cur_path/test/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log 2>&1 & done wait