File size: 263 Bytes
9cf79cf |
1 2 3 4 5 6 7 |
cd .. \
export CUDA_VISIBLE_DEVICES=0,1,2,3
torchrun --node_rank=0 --nproc_per_node=4 --nnodes=1 --master_addr="localhost" --master_port=12345 \
train_vbr_codec_rope.py \
-c /home/t2vg-a100-G4-10/project/qyp/mimc_rope/config/train_mim_config.yaml --world_size 4 |