VRIS_vip / scripts /dist_train_test_refcoco.sh
dianecy's picture
Add files using upload-large-folder tool
91e3dad verified
# example
python3 -m torch.distributed.launch --nproc_per_node=8 --use_env \
main_pretrain.py --with_box_refine --dataset_file all --batch_size 16 --num_frames 1 --binary \
--resume r50_pretrained.pth --eval
# --resume [path/to/weights.pth] --backbone [backbonetm]