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.
26 lines
589 B
26 lines
589 B
#!/bin/bash
|
|
|
|
ngpu=$1
|
|
feat_backend=$2
|
|
|
|
num_epochs=50
|
|
batch_size=16
|
|
ckpt_dir=./checkpoint
|
|
save_freq=10
|
|
|
|
if [ ${ngpu} -gt 0 ]; 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 \
|
|
--epochs ${num_epochs} \
|
|
--feat_backend ${feat_backend} \
|
|
--batch_size ${batch_size} \
|
|
--checkpoint_dir ${ckpt_dir} \
|
|
--save_freq ${save_freq}
|
|
fi
|