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

dataset_map_provider

Browse Source 
Summary: replace dataset_zoo with a pluggable DatasetMapProvider. The logic is now in annotated_file_dataset_map_provider.

Reviewed By: shapovalov

Differential Revision: D36443965

fbshipit-source-id: 9087649802810055e150b2fbfcc3c197a761f28a
This commit is contained in:
Jeremy Reizenstein
2022-05-20 07:50:30 -07:00
committed by Facebook GitHub Bot
parent 69c6d06ed8
commit 79c61a2d86
15 changed files with 305 additions and 175 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
projects/implicitron_trainer/configs/repro_base.yaml
View File

@@ -6,6 +6,7 @@ architecture: generic
visualize_interval: 0
visdom_port: 8097
data_source_args:
dataset_provider_class_type: JsonIndexDatasetMapProvider
dataloader_args:
batch_size: 10
dataset_len: 1000
@@ -21,7 +22,7 @@ data_source_args:
- 8
- 9
- 10
dataset_args:
dataset_map_provider_JsonIndexDatasetMapProvider_args:
dataset_root: ${oc.env:CO3D_DATASET_ROOT}
load_point_clouds: false
mask_depths: false