repro_singleseq_idr.yaml 1.21 KB
Newer Older
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
1
2
3
4
5
6
7
8
9
10
11
defaults:
- repro_singleseq_base
- _self_
generic_model_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
David Novotny's avatar
David Novotny committed
12
  view_pooler_enabled: false
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
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
  implicit_function_IdrFeatureField_args:
    n_harmonic_functions_xyz: 6
    bias: 0.6
    d_in: 3
    d_out: 1
    dims:
    - 512
    - 512
    - 512
    - 512
    - 512
    - 512
    - 512
    - 512
    geometric_init: true
    pooled_feature_dim: 0
    skip_in:
    - 6
    weight_norm: true
  renderer_SignedDistanceFunctionRenderer_args:
    ray_tracer_args:
      line_search_step: 0.5
      line_step_iters: 3
      n_secant_steps: 8
      n_steps: 100
      object_bounding_sphere: 8.0
      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
David Novotny's avatar
David Novotny committed
52
  raysampler_AdaptiveRaySampler_args:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
53
54
55
56
57
    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