forked from mindspore-Ecosystem/mindspore
fix retinanet shell script
This commit is contained in:
parent
ccee6de15f
commit
5d5b920227
|
@ -17,7 +17,7 @@
|
||||||
echo "=============================================================================================================="
|
echo "=============================================================================================================="
|
||||||
echo "Please run the script as: "
|
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 "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 "It is better to use absolute path."
|
||||||
echo "================================================================================================================="
|
echo "================================================================================================================="
|
||||||
|
|
||||||
|
|
|
@ -17,7 +17,7 @@
|
||||||
echo "=============================================================================================================="
|
echo "=============================================================================================================="
|
||||||
echo "Please run the script as: "
|
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 "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 "It is better to use absolute path."
|
||||||
echo "================================================================================================================="
|
echo "================================================================================================================="
|
||||||
|
|
||||||
|
@ -31,7 +31,6 @@ fi
|
||||||
# Before start single train, first create mindrecord files.
|
# Before start single train, first create mindrecord files.
|
||||||
BASE_PATH=$(cd "`dirname $0`" || exit; pwd)
|
BASE_PATH=$(cd "`dirname $0`" || exit; pwd)
|
||||||
cd $BASE_PATH/../ || exit
|
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"
|
echo "After running the script, the network runs in the background. The log will be generated in LOGx/log.txt"
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue