forked from CYang0515/NonCuboidRoom
-
Notifications
You must be signed in to change notification settings - Fork 0
/
cfg.yaml
61 lines (53 loc) · 914 Bytes
/
cfg.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
model_name: s3d
data: Structured3D
pretrained:
num_gpus: 4
batch_size: 24
num_workers: 16
exam: False
epochs: 51
lr_step: [30, 40]
Dataset:
Structured3D:
colorjitter: True
downsample: 4
input_h: 360
input_w: 640
max_objs: 20
SUNRGBD:
colorjitter: True
downsample: 4
input_h: 480
input_w: 640
max_objs: 20
NYU303:
colorjitter: True
downsample: 4
input_h: 480
input_w: 640
max_objs: 20
CUSTOM:
colorjitter: True
downsample: 4
input_h: 360
input_w: 640
max_objs: 20
Weights:
w_pcenter: 0.03
w_psize: 0.005
w_poffset: 0.02
w_loffset: 0.02
w_lalpha: 1
w_lregion: 0.03
w_pparam: 0.5
w_pparam_i: 0.5
w_pwdepth: 0.5
w_insdepth: 0.5
falpha_p: 2
fbeta_p: 4
falpha_l: 2
fbeta_l: 4
Solver:
method: adam
lr: 0.0001
weight_decay: 0.0001