Refactor autodecoders

Summary: Refactors autodecoders. Tests pass.

Reviewed By: bottler

Differential Revision: D37592429

fbshipit-source-id: 8f5c9eac254e1fdf0704d5ec5f69eb42f6225113

projects/implicitron_trainer/configs/repro_multiseq_idr_ad.yaml

@@ -11,10 +11,12 @@ generic_model_args:
   num_passes: 1
   output_rasterized_mc: true
   sampling_mode_training: mask_sample
-  sequence_autodecoder_args:
-    n_instances: 20000
-    init_scale: 1.0
-    encoding_dim: 256
+  global_encoder_class_type: SequenceAutodecoder
+  global_encoder_SequenceAutodecoder_args:
+    autodecoder_args:
+      n_instances: 20000
+      init_scale: 1.0
+      encoding_dim: 256
   implicit_function_IdrFeatureField_args:
     n_harmonic_functions_xyz: 6
     bias: 0.6

projects/implicitron_trainer/configs/repro_multiseq_nerf_ad.yaml

@@ -4,6 +4,8 @@ defaults:
 generic_model_args:
   chunk_size_grid: 16000
   view_pooler_enabled: false
-  sequence_autodecoder_args:
-    n_instances: 20000
-    encoding_dim: 256
+  global_encoder_class_type: SequenceAutodecoder
+  global_encoder_SequenceAutodecoder_args:
+    autodecoder_args:
+      n_instances: 20000
+      encoding_dim: 256

projects/implicitron_trainer/configs/repro_multiseq_srn_ad_hypernet.yaml

@@ -13,9 +13,11 @@ generic_model_args:
     loss_prev_stage_mask_bce: 0.0
     loss_autodecoder_norm: 0.001
     depth_neg_penalty: 10000.0
-  sequence_autodecoder_args:
-    encoding_dim: 256
-    n_instances: 20000
+  global_encoder_class_type: SequenceAutodecoder
+  global_encoder_SequenceAutodecoder_args:
+    autodecoder_args:
+      encoding_dim: 256
+      n_instances: 20000
   raysampler_class_type: NearFarRaySampler
   raysampler_NearFarRaySampler_args:
     n_rays_per_image_sampled_from_mask: 2048

projects/implicitron_trainer/tests/experiment.yaml

@@ -16,6 +16,7 @@ generic_model_args:
   n_train_target_views: 1
   sampling_mode_training: mask_sample
   sampling_mode_evaluation: full_grid
+  global_encoder_class_type: null
   raysampler_class_type: AdaptiveRaySampler
   renderer_class_type: MultiPassEmissionAbsorptionRenderer
   image_feature_extractor_class_type: null
@@ -49,11 +50,16 @@ generic_model_args:
   - objective
   - epoch
   - sec/it
-  sequence_autodecoder_args:
-    encoding_dim: 0
-    n_instances: 0
-    init_scale: 1.0
-    ignore_input: false
+  global_encoder_HarmonicTimeEncoder_args:
+    n_harmonic_functions: 10
+    append_input: true
+    time_divisor: 1.0
+  global_encoder_SequenceAutodecoder_args:
+    autodecoder_args:
+      encoding_dim: 0
+      n_instances: 0
+      init_scale: 1.0
+      ignore_input: false
   raysampler_AdaptiveRaySampler_args:
     image_width: 400
     image_height: 400