9cf79cf
1
2
3
4
5
6
7
cd .. \ export CUDA_VISIBLE_DEVICES=0 torchrun --node_rank=0 --nproc_per_node=1 --nnodes=1 --master_addr="localhost" --master_port=12345 \ train_vbr_codec.py \ -c /home/v-ruoyufeng/v-ruoyufeng/qyp/mage/config/train_mim_config.yaml --world_size 1
