VoxCPM-BACKUP_2 / scripts /example_train.sh
Jiwon-Kang's picture
Add files using upload-large-folder tool
bcc1595 verified
Raw
History Blame Contribute Delete
528 Bytes
# Single GPU
# python scripts/train_voxcpm_finetune.py --config_path conf/librispeech_full.yaml
# Multi-GPU (4×)
cd /data/mm-llm-backbone_890/personal/sirius/VoxCPM
source ./.venv/bin/activate
source env.sh
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/TEST_EXP_ffn6144_ditNum4.yaml