NUM_GPUS=${1:-1}  # default: 1 GPU

BATCH_SIZE=$((30 / NUM_GPUS))

echo "Using $NUM_GPUS GPUs, each with a batch size of $BATCH_SIZE"

accelerate launch --num_processes $NUM_GPUS train_motionstreamer.py \
--batch-size $BATCH_SIZE \
--lr 0.0001 \
--total-iter 100000 \
--out-dir Experiments \
--exp-name motionstreamer_model \
--dataname t2m_babel_272 \
--latent_dir babel_272_stream/t2m_babel_latents \
--num_gpus $NUM_GPUS