-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathsoma_aood_pascal_to_clipart_r50.yaml
93 lines (93 loc) · 1.92 KB
/
soma_aood_pascal_to_clipart_r50.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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
EVAL_EPOCH: 1
CACHE_MODE: False
AOOD:
MOTIF_ON: True
# novel-class
OPEN_SET:
WARM_UP: 9
TH: 1.0
MOTIF_ON: True
ALPHA: 0.01
KNN: 5
UNK_PROB: 0.5
MOTIF_LOSS_COEF: 0.01
# novel-scene
CROSS_DOMAIN:
MOTIF_ON: True
BETA: 2.0 # std scaling
DATASET:
DA_MODE: aood
AOOD_SETTING: 1
AOOD_SCENE: 'pascal'
COCO_PANOPTIC_PATH: None
COCO_PATH: /home/wuyangli2/data/
DATASET_FILE: pascal_to_clipart
NUM_CLASSES: 11 # 10 known + 1
REMOVE_DIFFICULT: False
DEVICE: cuda
DIST:
DISTRIBUTED: False
DIST_BACKEND: nccl
DIST_URL: env://
GPU: 0
RANK: 0
WORLD_SIZE: 4
EVAL: False
LOSS:
AUX_LOSS: True
BACKBONE_LOSS_COEF: 0.1
BBOX_LOSS_COEF: 5.0
CHANNEL_QUERY_LOSS_COEF: 0.1
CLS_LOSS_COEF: 2.0
DA_GAMMA: 0
DICE_LOSS_COEF: 1.0
FOCAL_ALPHA: 0.25
GIOU_LOSS_COEF: 2.0
INSTANCE_QUERY_LOSS_COEF: 0.1
MASK_LOSS_COEF: 1.0
SET_COST_BBOX: 5.0
SET_COST_CLASS: 2.0
SET_COST_GIOU: 2.0
SPACE_QUERY_LOSS_COEF: 0.1
MODEL:
BACKBONE: resnet50
BACKBONE_ALIGN: True
CHANNEL_ALIGN: False
DEC_LAYERS: 3
DEC_N_POINTS: 4
DILATION: False
DIM_FEEDFORWARD: 1024
DROPOUT: 0.1
ENC_LAYERS: 3
ENC_N_POINTS: 4
FROZEN_WEIGHTS: None
HIDDEN_DIM: 256
INSTANCE_ALIGN: False
MASKS: False
NHEADS: 8
NUM_FEATURE_LEVELS: 4
NUM_QUERIES: 100
POSITION_EMBEDDING: sine
POSITION_EMBEDDING_SCALE: 6.283185307179586
SPACE_ALIGN: False
TWO_STAGE: False
WITH_BOX_REFINE: False
NUM_WORKERS: 2
OUTPUT_DIR: exps/r50_aood_p2c
RESUME:
LOAD_OPTIMIZER: False
SEED: 42 # This cannot strictly control the same results. Sorry, I do not know why.
START_EPOCH: 0
TRAIN:
BATCH_SIZE: 4 # each gpu, 2 for source and 2 for target
CLIP_MAX_NORM: 0.1
EPOCHS: 30
LR: 0.0002
LR_BACKBONE: 2e-05
LR_BACKBONE_NAMES: ['backbone.0']
LR_DROP: 25
LR_DROP_EPOCHS: None
LR_LINEAR_PROJ_MULT: 0.1
LR_LINEAR_PROJ_NAMES: ['reference_points', 'sampling_offsets']
SGD: False
WEIGHT_DECAY: 0.0001