# Single GPU
# python scripts/train_voxcpm_finetune.py --config_path conf/librispeech_full_voxcpm_v1_whisper.yaml

# Multi-GPU (2×)
source ./.venv/bin/activate
export TORCH_NCCL_HEARTBEAT_TIMEOUT_SEC="${TORCH_NCCL_HEARTBEAT_TIMEOUT_SEC:-3600}"
# print python env info
echo "Python executable: $(which python)"
CUDA_VISIBLE_DEVICES=0,1 torchrun --nproc_per_node=2 \
    scripts/train_voxcpm_finetune.py --config_path conf/librispeech_full_voxcpm_v1_whisper.yaml