mimc_rope / scripts /train_det.sh
Qiyp's picture
add random cls ckpt
042a375
cd .. \
export CUDA_VISIBLE_DEVICES=0
torchrun --node_rank=0 --nproc_per_node=1 --nnodes=1 --master_addr="localhost" --master_port=12345 \
train_on_coco.py \
-c /home/v-ruoyufeng/v-ruoyufeng/qyp/mage/config/train_det.yaml --world_size 1