p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

ssd_ghost gpu standalone bug fix

This commit is contained in:
zhangxiaoxiao 2021-07-26 11:51:48 +08:00
parent 305f08811d
commit 3ff0609734
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
model_zoo/research/cv/ssd_ghostnet/scripts/run_standalone_train_gpu.sh
View File

@ -42,7 +42,7 @@ CONFIG_PATH=$5
PRE_TRAINED=$6
PRE_TRAINED_EPOCH_SIZE=$7
export DEVICE_ID=$DEVICE_ID
export CUDA_VISIBLE_DEVICES=$DEVICE_ID
rm -rf LOG$DEVICE_ID
mkdir ./LOG$DEVICE_ID
cp ./*.py ./LOG$DEVICE_ID
@ -56,7 +56,7 @@ then
python train.py \
--lr=$LR \
--dataset=$DATASET \
--device_id=$DEVICE_ID \
--device_id=0 \
--loss_scale=1 \
--device_target="GPU" \
--config_path=$CONFIG_PATH \
@ -68,7 +68,7 @@ then
python train.py \
--lr=$LR \
--dataset=$DATASET \
--device_id=$DEVICE_ID \
--device_id=0 \
--loss_scale=1 \
--device_target="GPU" \
--pre_trained=$PRE_TRAINED \