mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-12-16 04:10:35 +08:00
CO3Dv2 trainer configs
Summary: Adds yaml configs to train selected methods on CO3Dv2. Few more updates: 1) moved some fields to base classes so that we can check is_multisequence in experiment.py 2) skip loading all train cameras for multisequence datasets, without this, co3d-fewview is untrainable 3) fix bug in json index dataset provider v2 Reviewed By: kjchalup Differential Revision: D38952755 fbshipit-source-id: 3edac6fc8e20775aa70400bd73a0e6d52b091e0c
This commit is contained in:
committed by
Facebook GitHub Bot
parent
03562d87f5
commit
1163eaab43
@@ -0,0 +1,8 @@
|
||||
data_source_ImplicitronDataSource_args:
|
||||
dataset_map_provider_class_type: JsonIndexDatasetMapProviderV2
|
||||
dataset_map_provider_JsonIndexDatasetMapProviderV2_args:
|
||||
category: teddybear
|
||||
subset_name: fewview_dev
|
||||
training_loop_ImplicitronTrainingLoop_args:
|
||||
evaluator_ImplicitronEvaluator_args:
|
||||
is_multisequence: true
|
||||
@@ -0,0 +1,4 @@
|
||||
defaults:
|
||||
- repro_multiseq_nerf_wce.yaml
|
||||
- repro_multiseq_co3dv2_base.yaml
|
||||
- _self_
|
||||
@@ -0,0 +1,4 @@
|
||||
defaults:
|
||||
- repro_multiseq_nerformer.yaml
|
||||
- repro_multiseq_co3dv2_base.yaml
|
||||
- _self_
|
||||
@@ -0,0 +1,4 @@
|
||||
defaults:
|
||||
- repro_multiseq_srn_ad_hypernet.yaml
|
||||
- repro_multiseq_co3dv2_base.yaml
|
||||
- _self_
|
||||
@@ -0,0 +1,4 @@
|
||||
defaults:
|
||||
- repro_multiseq_srn_wce.yaml
|
||||
- repro_multiseq_co3dv2_base.yaml
|
||||
- _self_
|
||||
@@ -0,0 +1,8 @@
|
||||
data_source_ImplicitronDataSource_args:
|
||||
dataset_map_provider_class_type: JsonIndexDatasetMapProviderV2
|
||||
dataset_map_provider_JsonIndexDatasetMapProviderV2_args:
|
||||
category: teddybear
|
||||
subset_name: manyview_dev_0
|
||||
training_loop_ImplicitronTrainingLoop_args:
|
||||
evaluator_ImplicitronEvaluator_args:
|
||||
is_multisequence: false
|
||||
@@ -0,0 +1,4 @@
|
||||
defaults:
|
||||
- repro_singleseq_idr.yaml
|
||||
- repro_singleseq_co3dv2_base.yaml
|
||||
- _self_
|
||||
@@ -0,0 +1,4 @@
|
||||
defaults:
|
||||
- repro_singleseq_nerf.yaml
|
||||
- repro_singleseq_co3dv2_base.yaml
|
||||
- _self_
|
||||
@@ -0,0 +1,4 @@
|
||||
defaults:
|
||||
- repro_singleseq_nerformer.yaml
|
||||
- repro_singleseq_co3dv2_base.yaml
|
||||
- _self_
|
||||
@@ -0,0 +1,4 @@
|
||||
defaults:
|
||||
- repro_singleseq_srn_noharm.yaml
|
||||
- repro_singleseq_co3dv2_base.yaml
|
||||
- _self_
|
||||
Reference in New Issue
Block a user