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.
28 lines
624 B
28 lines
624 B
3 years ago
|
#!/bin/bash
|
||
|
|
|
||
|
|
ngpu=$1
|
||
|
|
device=$2
|
||
|
|
feat_backend=$3
|
||
|
|
|
||
|
|
num_epochs=50
|
||
|
|
batch_size=16
|
||
|
|
ckpt_dir=./checkpoint
|
||
|
|
save_freq=10
|
||
|
|
|
||
|
|
if [ ${ngpu} -gt 1 ]; then
|
||
|
|
python3 -m paddle.distributed.launch --gpus $CUDA_VISIBLE_DEVICES ${BIN_DIR}/train.py \
|
||
|
|
--epochs ${num_epochs} \
|
||
|
|
--feat_backend ${feat_backend} \
|
||
|
|
--batch_size ${batch_size} \
|
||
|
|
--checkpoint_dir ${ckpt_dir} \
|
||
|
|
--save_freq ${save_freq}
|
||
|
|
else
|
||
|
|
python3 ${BIN_DIR}/train.py \
|
||
|
|
--device ${device} \
|
||
|
|
--epochs ${num_epochs} \
|
||
|
|
--feat_backend ${feat_backend} \
|
||
|
|
--batch_size ${batch_size} \
|
||
|
|
--checkpoint_dir ${ckpt_dir} \
|
||
|
|
--save_freq ${save_freq}
|
||
|
|
fi
|