| 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 |