423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-14 19:36:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor ViewMetrics

Browse Source 
Summary:
Make ViewMetrics easy to replace by putting them into an OmegaConf dataclass.

Also, re-word a few variable names and fix minor TODOs.

Reviewed By: bottler

Differential Revision: D37327157

fbshipit-source-id: 78d8e39bbb3548b952f10abbe05688409fb987cc
This commit is contained in:
Krzysztof Chalupka
2022-06-30 09:22:01 -07:00
committed by Facebook GitHub Bot
parent f4dd151037
commit ae35824f21
5 changed files with 301 additions and 151 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
projects/implicitron_trainer/tests/experiment.yaml
View File

@@ -21,6 +21,8 @@ generic_model_args:
image_feature_extractor_class_type: null
view_pooler_enabled: false
implicit_function_class_type: NeuralRadianceFieldImplicitFunction
view_metrics_class_type: ViewMetrics
regularization_metrics_class_type: RegularizationMetrics
loss_weights:
loss_rgb_mse: 1.0
loss_prev_stage_rgb_mse: 1.0
@@ -268,6 +270,8 @@ generic_model_args:
in_features: 256
out_features: 3
ray_dir_in_camera_coords: false
view_metrics_ViewMetrics_args: {}
regularization_metrics_RegularizationMetrics_args: {}
solver_args:
breed: adam
weight_decay: 0.0