mimc_rl / scripts /train_codec_rope.sh
wangyanhui666's picture
fine tune decoder with mask
9cf79cf
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