Uni-hema / scripts /DINO_train_submitit.sh
ryhm's picture
Upload 206 files
51067bb verified
Raw
History Blame Contribute Delete
307 Bytes
coco_path=$1
python run_with_submitit.py --timeout 3000 --job_name DINO \
--job_dir logs/DINO/R50-MS4-%j --ngpus 8 --nodes 1 \
-c config/DINO/DINO_4scale.py --coco_path $coco_path \
--options dn_scalar=100 embed_init_tgt=TRUE \
dn_label_coef=1.0 dn_bbox_coef=1.0 use_ema=False \
dn_box_noise_scale=1.0