From 5d5b92022771813fcdd3001dcd408b3a069ead5a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=99=88=E5=8A=A2?= Date: Tue, 1 Jun 2021 10:56:47 +0800 Subject: [PATCH] fix retinanet shell script --- .../official/cv/retinanet/scripts/run_distribute_train.sh | 2 +- model_zoo/official/cv/retinanet/scripts/run_single_train.sh | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/model_zoo/official/cv/retinanet/scripts/run_distribute_train.sh b/model_zoo/official/cv/retinanet/scripts/run_distribute_train.sh index efd860b697c..26560f80d69 100644 --- a/model_zoo/official/cv/retinanet/scripts/run_distribute_train.sh +++ b/model_zoo/official/cv/retinanet/scripts/run_distribute_train.sh @@ -17,7 +17,7 @@ echo "==============================================================================================================" echo "Please run the script as: " echo "sh run_distribute_train.sh DEVICE_NUM EPOCH_SIZE LR DATASET RANK_TABLE_FILE PRE_TRAINED PRE_TRAINED_EPOCH_SIZE" -echo "for example: sh run_distribute_train.sh 8 500 0.1 coco /data/hccl.json /opt/retinanet-500_458.ckpt(optional) 200(optional)" +echo "for example: sh run_distribute_train.sh 8 500 0.1 /data/hccl.json /opt/retinanet-500_458.ckpt(optional) 200(optional)" echo "It is better to use absolute path." echo "=================================================================================================================" diff --git a/model_zoo/official/cv/retinanet/scripts/run_single_train.sh b/model_zoo/official/cv/retinanet/scripts/run_single_train.sh index 53f998df057..3c7cd36ce4a 100644 --- a/model_zoo/official/cv/retinanet/scripts/run_single_train.sh +++ b/model_zoo/official/cv/retinanet/scripts/run_single_train.sh @@ -17,7 +17,7 @@ echo "==============================================================================================================" echo "Please run the script as: " echo "sh run_single_train.sh DEVICE_ID EPOCH_SIZE LR DATASET PRE_TRAINED PRE_TRAINED_EPOCH_SIZE" -echo "for example: sh run_single_train.sh 0 500 0.1 coco /opt/retinanet-500_458.ckpt(optional) 200(optional)" +echo "for example: sh run_single_train.sh 0 500 0.1 /opt/retinanet-500_458.ckpt(optional) 200(optional)" echo "It is better to use absolute path." echo "=================================================================================================================" @@ -31,7 +31,6 @@ fi # Before start single train, first create mindrecord files. BASE_PATH=$(cd "`dirname $0`" || exit; pwd) cd $BASE_PATH/../ || exit -python train.py --only_create_dataset=True echo "After running the script, the network runs in the background. The log will be generated in LOGx/log.txt"