Update hoidet_hico_dla_glob_3level.sh

staceycy · Mar 22, 2020 · ef85718 · ef85718
1 parent 96b4cc3
commit ef85718
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/experiments/hoidet_hico_dla_glob_3level.sh b/experiments/hoidet_hico_dla_glob_3level.sh
@@ -6,7 +6,7 @@ set -x
 cd src
 
 #train
-python -u  main.py hoidet --exp_id hoidet_hico_dla_3levelglob --arch dla3levelglob_34 --batch_size 112 --master_batch 7  --lr 4.5e-4 --gpus 0,1,2,3,4,5,6,7 --num_workers 16 --val_intervals 100000 --load_model ../models/ctdet_coco_dla_2x.pth --image_dir images/train2015 --dataset hico --root_path '/mnt/lustre/liaoyue/datasets'
+python -u  main.py hoidet --exp_id hoidet_hico_dla_3levelglob --arch dla3levelglob_34 --batch_size 104 --master_batch 6  --lr 4e-4 --gpus 0,1,2,3,4,5,6,7 --num_workers 16 --val_intervals 100000 --load_model ../models/ctdet_coco_dla_2x.pth --image_dir images/train2015 --dataset hico --root_path '/mnt/lustre/liaoyue/datasets'
 
 #test
 python -u test_hoi.py hoidet --exp_id hoidet_hico_dla_3levelglob --arch dla3levelglob_34 --gpus 0 --image_dir images/test2015 --dataset hico --test_with_eval