From 6636d245144aa029af56c78ea451fa32982b8a3a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=9B=BD=E5=BA=86?= <3224925783@qq.com> Date: Wed, 23 Nov 2022 07:12:02 +0000 Subject: [PATCH] update TensorFlow/contrib/cv/DANN_ID2200_for_TensorFlow/test/train_performance_1p.sh. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 王国庆 <3224925783@qq.com> --- .../test/train_performance_1p.sh | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/TensorFlow/contrib/cv/DANN_ID2200_for_TensorFlow/test/train_performance_1p.sh b/TensorFlow/contrib/cv/DANN_ID2200_for_TensorFlow/test/train_performance_1p.sh index ee5bb1b9e..7cb47c9eb 100644 --- a/TensorFlow/contrib/cv/DANN_ID2200_for_TensorFlow/test/train_performance_1p.sh +++ b/TensorFlow/contrib/cv/DANN_ID2200_for_TensorFlow/test/train_performance_1p.sh @@ -96,6 +96,8 @@ start_time=$(date +%s) #进入训练脚本目录，需要模型审视修改 cd $cur_path/ +sed -i "s|"MNIST_data"|"${data_path}/MNIST"|g" MNIST-DANN.py +sed -i "s|"mnistm_data.pkl"|"${data_path}/MNIST/mnistm_data.pkl"|g" MNIST-DANN.py for((RANK_ID=$RANK_ID_START;RANK_ID<$((RANK_SIZE+RANK_ID_START));RANK_ID++)); do #设置环境变量，不需要修改 @@ -123,15 +125,10 @@ do #执行训练脚本，以下传参不需要修改，其他需要模型审视修改 #--data_dir, --model_dir, --precision_mode, --over_dump, --over_dump_path，--data_dump_flag，--data_dump_step，--data_dump_path，--profiling，--profiling_dump_path - sed -i "s|"MNIST_data"|"${data_path}/MNIST"|g" MNIST-DANN.py - sed -i "s|"mnistm_data.pkl"|"${data_path}/MNIST/mnistm_data.pkl"|g" MNIST-DANN.py python3 MNIST-DANN.py \ --num_steps 1000 > ${cur_path}test/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & - - sed -i "s|"${data_path}/MNIST/mnistm_data.pkl"|"mnistm_data.pkl"|g" MNIST-DANN.py - sed -i "s|"${data_path}/MNIST"|"MNIST_data"|g" MNIST-DANN.py done wait @@ -139,6 +136,9 @@ wait end_time=$(date +%s) e2e_time=$(( $end_time - $start_time )) +sed -i "s|"${data_path}/MNIST/mnistm_data.pkl"|"mnistm_data.pkl"|g" MNIST-DANN.py +sed -i "s|"${data_path}/MNIST"|"MNIST_data"|g" MNIST-DANN.py + #结果打印，不需要修改 echo "------------------ Final result ------------------" #输出性能FPS，需要模型审视修改 -- Gitee