#!/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