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

more padding options in Dataloader

Browse Source 
Summary: Add facilities for dataloading non-sequential scenes.

Reviewed By: shapovalov

Differential Revision: D37291277

fbshipit-source-id: 0a33e3727b44c4f0cba3a2abe9b12f40d2a20447
This commit is contained in:
Jeremy Reizenstein
2022-07-06 07:13:41 -07:00
committed by Facebook GitHub Bot
parent 0dce883241
commit 771cf8a328
16 changed files with 449 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -6,22 +6,12 @@ architecture: generic
visualize_interval: 0
visdom_port: 8097
data_source_args:
data_loader_map_provider_class_type: SequenceDataLoaderMapProvider
dataset_map_provider_class_type: JsonIndexDatasetMapProvider
data_loader_map_provider_SequenceDataLoaderMapProvider_args:
batch_size: 10
dataset_len: 1000
dataset_len_val: 1
dataset_length_train: 1000
dataset_length_val: 1
num_workers: 8
images_per_seq_options:
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
dataset_map_provider_JsonIndexDatasetMapProvider_args:
dataset_root: ${oc.env:CO3D_DATASET_ROOT}
n_frames_per_sequence: -1

4
projects/implicitron_trainer/configs/repro_multiseq_base.yaml
View File

@@ -4,8 +4,8 @@ defaults:
data_source_args:
data_loader_map_provider_SequenceDataLoaderMapProvider_args:
batch_size: 10
dataset_len: 1000
dataset_len_val: 1
dataset_length_train: 1000
dataset_length_val: 1
num_workers: 8
images_per_seq_options:
- 2

6
projects/implicitron_trainer/configs/repro_singleseq_base.yaml
View File

@@ -4,11 +4,9 @@ defaults:
data_source_args:
data_loader_map_provider_SequenceDataLoaderMapProvider_args:
batch_size: 1
dataset_len: 1000
dataset_len_val: 1
dataset_length_train: 1000
dataset_length_val: 1
num_workers: 8
images_per_seq_options:
- 2
dataset_map_provider_JsonIndexDatasetMapProvider_args:
assert_single_seq: true
n_frames_per_sequence: -1

4
projects/implicitron_trainer/configs/repro_singleseq_wce_base.yaml
View File

@@ -4,8 +4,8 @@ defaults:
data_source_args:
data_loader_map_provider_SequenceDataLoaderMapProvider_args:
batch_size: 10
dataset_len: 1000
dataset_len_val: 1
dataset_length_train: 1000
dataset_length_val: 1
num_workers: 8
images_per_seq_options:
- 2