repro_singleseq_idr.yaml 1.32 KB
Newer Older
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
1
2
3
defaults:
- repro_singleseq_base
- _self_
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
model_factory_ImplicitronModelFactory_args:
  model_GenericModel_args:
    loss_weights:
      loss_mask_bce: 100.0
      loss_kl: 0.0
      loss_rgb_mse: 1.0
      loss_eikonal: 0.1
    chunk_size_grid: 65536
    num_passes: 1
    view_pooler_enabled: false
    implicit_function_IdrFeatureField_args:
      n_harmonic_functions_xyz: 6
      bias: 0.6
      d_in: 3
      d_out: 1
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
19
20
21
22
23
      dims:
      - 512
      - 512
      - 512
      - 512
24
25
26
27
28
      - 512
      - 512
      - 512
      - 512
      geometric_init: true
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
29
      pooled_feature_dim: 0
30
31
      skip_in:
      - 6
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
32
      weight_norm: true
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
    renderer_SignedDistanceFunctionRenderer_args:
      ray_tracer_args:
        line_search_step: 0.5
        line_step_iters: 3
        n_secant_steps: 8
        n_steps: 100
        sdf_threshold: 5.0e-05
      ray_normal_coloring_network_args:
        d_in: 9
        d_out: 3
        dims:
        - 512
        - 512
        - 512
        - 512
        mode: idr
        n_harmonic_functions_dir: 4
        pooled_feature_dim: 0
        weight_norm: true
    raysampler_AdaptiveRaySampler_args:
      n_rays_per_image_sampled_from_mask: 1024
      n_pts_per_ray_training: 0
      n_pts_per_ray_evaluation: 0
    renderer_class_type: SignedDistanceFunctionRenderer
    implicit_function_class_type: IdrFeatureField