You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
13 lines
239 B
13 lines
239 B
2 years ago
|
#!/bin/bash
|
||
|
|
|
||
|
|
ngpu=$1
|
||
|
|
cfg_path=$2
|
||
|
|
|
||
|
|
if [ ${ngpu} -gt 0 ]; then
|
||
|
|
python3 -m paddle.distributed.launch --gpus $CUDA_VISIBLE_DEVICES local/train.py \
|
||
|
|
--cfg_path ${cfg_path}
|
||
|
|
else
|
||
|
|
python3 local/train.py \
|
||
|
|
--cfg_path ${cfg_path}
|
||
|
|
fi
|