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