-
Notifications
You must be signed in to change notification settings - Fork 1
/
train.sh
19 lines (14 loc) · 1.52 KB
/
train.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
################Deformable Detr###################
CUDA_VISIBLE_DEVICES=0,1,2,3 bash ./tools/dist_train.sh configs/deformable_detr/FD/deformable_detr_FD_load.py 4 --work-dir ./experiments/FD_load_deform --auto-scale-lr
###################DINO############################
CUDA_VISIBLE_DEVICES=0,1,2,3 bash ./tools/dist_train.sh configs/dino/FD/dino_detr_FD_load.py 4 --work-dir ./experiments/FD_load_dino --auto-scale-lr
##############DIFF DETR############################
CUDA_VISIBLE_DEVICES=0,1,2,3 bash ./tools/dist_train.sh configs/DiffusionDet/configs/FD/diff_detr_FD_load.py 4 --work-dir ./experiments/FD_load_diff --auto-scale-lr
###################GLIP EXPERIMENTS################
CUDA_VISIBLE_DEVICES=0,1,2,3 bash ./tools/dist_train.sh configs/glip/FD/glip_FD_baseline.py 4 --work-dir ./experiments/FD_glip --auto-scale-lr
CUDA_VISIBLE_DEVICES=0,1,2,3 bash ./tools/dist_train.sh configs/glip/FD/glip_FD_BCG.py 4 --work-dir ./experiments/FD_BCG_glip --auto-scale-lr
####################Grounding DINO Baseline########
CUDA_VISIBLE_DEVICES=0,1,2,3 bash ./tools/dist_train.sh configs/grounding_dino/FD/grounding_FD_baseline.py 4 --work-dir ./experiments/FD_gdino --auto-scale-lr
CUDA_VISIBLE_DEVICES=0,1,2,3 bash ./tools/dist_train.sh configs/grounding_dino/FD/grounding_FD_BCG.py 4 --work-dir ./experiments/FD_BCG_gdino --auto-scale-lr
#####################FD-SOS########################
CUDA_VISIBLE_DEVICES=0,1,2,3 bash ./tools/dist_train.sh configs/FD-SOS/grounding_FD_BCG_teeth_specific.py 4 --work-dir ./experiments/FD_BCG_SOS --auto-scale-lr