mimc_rl / scripts /eval_only.sh
wangyanhui666's picture
fine tune decoder with mask
9cf79cf
cd .. \
export CUDA_VISIBLE_DEVICES=3
torchrun --node_rank=0 --nproc_per_node=1 --nnodes=1 --master_addr="localhost" --master_port=12346 \
eval_only.py \
-c /home/t2vg-a100-G4-10/project/qyp/mimc_rope/config/eval_high_res.yaml --world_size 1
# torchrun --node_rank=0 --nproc_per_node=1 --nnodes=1 --master_addr="localhost" --master_port=12345 \
# test_inference.py \
# -c /home/t2vg-a100-G4-10/project/qyp/mimc_rope/config/train_det.yaml --world_size 1