/
replica-room0.yaml
75 lines (72 loc) · 1.37 KB
/
replica-room0.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
name: 'replica-room0'
notes: '
[Data] room_0
[Model] 500 planes
[init] dense points (w/ box)
[Loss] mse_color + mse_point2plane + area_reg.
[Reg.] oscillate NDC
'
seed: 0
cuda: '0'
data:
path: 'data/replica/room_0/'
image_size: [512, 512]
sample_rate: 1.0
batch_points: 10000
white_bg: False
checkpoint:
teacher: 'replica-room0-teacher.pth'
experts: 'replica-room0-experts.pth'
model:
n_plane: 500
n_train_sample: 4096
n_infer_sample: 4096
anti_aliasing: True
premultiply_alpha: False
init:
lrf_neighbors: 100
wh: 2
box_factor: 4
random_rate: 0.0
mlp_teacher:
n_harmonic_functions_pos: 10
n_harmonic_functions_dir: 4
n_hidden_neurons_pos: 256
n_hidden_neurons_dir: 128
n_layers: 8
mlp_experts:
n_harmonic_functions_pos: 10
n_harmonic_functions_dir: 4
n_hidden_neurons_pos: 32
n_hidden_neurons_dir: 32
n_layers: 2
accelerate:
bake: False
n_bake_sample: 10000
bake_res: 200
thresh: 0.001
loss_weight:
color: 1.0
point2plane: 1.0
area: 0.00000001
optimizer:
lr: 0.0005
lr_scheduler_step_size: 5000
lr_scheduler_gamma: 0.1
train:
resume: True
freeze_geometry: True
epoch:
teacher: 2000
distill: 1000
finetune: 1000
validation: 500
checkpoint: 20
test:
mode: 'test_model'
folder: ''
vis:
id: ''
r: 3
c: [0.5, 0.5, 0.5]
alpha: 0.3