-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathfern.yml
94 lines (92 loc) · 2.16 KB
/
fern.yml
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
94
datasets:
- type: LLFFDataset
base_dir: "data/nerf_llff/fern"
split: "train"
- type: LLFFDataset
base_dir: "data/nerf_llff/fern"
split: "val"
- type: LLFFDataset
base_dir: "data/nerf_llff/fern"
split: "test"
runner:
eval_last_epoch_model: true
seed: 42
output_dir: results/nerf/fern
debug: False
print_per_iter: 100
val_per_iter: 20000
save_per_iter: 10000
init_lr: 5.0e-4
weight_decay: 0.0
# lr warmup
warmup_steps: 1000
warmup_lr: 1.0e-5
linear_scale: True
# lr decay
lr_decay_type: exponential # cosine, exponential
min_lr: 5.0e-5
# 0.99..99: num_decimal_epoch * num_decimal_rate = 1, rate ** num_epoch ~= 0.36
# 0.99..95: num_decimal_epoch * num_decimal_rate = 0.1, rate ** num_epoch ~= 0.60
lr_decay_rate: 0.1
lr_decay_iters: 250000
batch_size_list:
- 1
- 1
- 1
num_workers_list:
- 4
- 4
- 4
num_iters: 200000
hooks: []
lr_param_groups: []
pipeline:
type: NeRFPipeline
chunk_size_grid: 131072
num_passes: 2
output_rasterized_mc: true
loss_weights:
loss_prev_stage_rgb_mse: 1.0
loss_rgb_mse: 1.0
model:
type: NeRFMLP
color_dim: 3
harmonic_functions_dir_append_intput: true
harmonic_functions_xyz_append_intput: true
input_dir: true
input_skips:
- 5
input_xyz: true
latent_dim: 0
n_harmonic_functions_dir: 4
n_harmonic_functions_xyz: 10
n_hidden_neurons_dir: 128
n_hidden_neurons_xyz: 256
n_layers: 8
nerf_paper_v1: false
ray_sampler:
type: RaySampler
image_height: 378
image_width: 504
min_depth: 2.0
max_depth: 6.0
n_pts_per_ray_evaluation: 64
n_pts_per_ray_training: 64
n_rays_per_image_sampled_from_mask: 1024
scene_extent: 0.0
stratified_point_sampling_training: true
stratified_point_sampling_evaluation: false
renderer:
type: MultipassEmissionAbsorpsionRenderer
append_coarse_samples_to_fine: true
bg_color:
- 0.0
- 0.0
- 0.0
blend_output: false
density_noise_std_train: 0.0
n_pts_per_ray_fine_evaluation: 64
n_pts_per_ray_fine_training: 64
hard_background: false
background_density_bias: 1.0e-6
feature_extractor: []