-
Notifications
You must be signed in to change notification settings - Fork 9
/
config.yaml
41 lines (36 loc) · 769 Bytes
/
config.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# BASE CONFIG
EXP: # folder structure will be ROOT/NAME_DATE_TIME
ROOT: output # /path/to/experiment_output
NAME: instr
DATA:
NUM_WORKERS: 4
TRAIN:
ROOT: /path/to/train_split_folder
BATCH_SIZE: 2
TRANSFORMS: True
VAL:
ROOT: /path/to/val_split_folder
BATCH_SIZE: 2
TRANSFORMS: False
TEST:
ROOT: /path/to/stios
BATCH_SIZE: 2
TRANSFORMS: False
MODEL:
NUM_QUERIES: 15
AUX_DECODER_LOSS: True
WITH_DISP: True
AXIAL_ATTENTION: True
QUERY_PROC: expanded # expanded, att, attcat_tfenc, attcat_bb
LOSS:
POWER: 0.2
POS_WEIGHT: 1.
NEG_WEIGHT: 1.
OPTIMIZER:
TYPE: "optim.AdamW" # AdamW is default
LR: 0.0001
WEIGHT_DECAY: 0.01
SCHEDULER:
TYPE: "optim.lr_scheduler.StepLR"
step_size: 10
gamma: 0.5