repro_singleseq_srn.yaml 740 Bytes
Newer Older
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
1
2
3
4
5
6
defaults:
- repro_singleseq_base.yaml
- _self_
generic_model_args:
  num_passes: 1
  chunk_size_grid: 32000
David Novotny's avatar
David Novotny committed
7
  view_pooler_enabled: false
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
  loss_weights:
    loss_rgb_mse: 200.0
    loss_prev_stage_rgb_mse: 0.0
    loss_mask_bce: 1.0
    loss_prev_stage_mask_bce: 0.0
    loss_autodecoder_norm: 0.0
    depth_neg_penalty: 10000.0
  raysampler_args:
    n_rays_per_image_sampled_from_mask: 2048
    min_depth: 0.05
    max_depth: 0.05
    scene_extent: 0.0
    n_pts_per_ray_training: 1
    n_pts_per_ray_evaluation: 1
    stratified_point_sampling_training: false
    stratified_point_sampling_evaluation: false
  renderer_class_type: LSTMRenderer
  implicit_function_class_type: SRNImplicitFunction
solver_args:
  breed: adam
  lr: 5.0e-05