diff --git a/projects/implicitron_trainer/configs/repro_base.yaml b/projects/implicitron_trainer/configs/repro_base.yaml index 041ba47b..1cf08f07 100644 --- a/projects/implicitron_trainer/configs/repro_base.yaml +++ b/projects/implicitron_trainer/configs/repro_base.yaml @@ -6,7 +6,7 @@ architecture: generic visualize_interval: 0 visdom_port: 8097 data_source_args: - dataset_provider_class_type: JsonIndexDatasetMapProvider + dataset_map_provider_class_type: JsonIndexDatasetMapProvider data_loader_map_provider_SequenceDataLoaderMapProvider_args: batch_size: 10 dataset_len: 1000 diff --git a/projects/implicitron_trainer/configs/repro_singleseq_base.yaml b/projects/implicitron_trainer/configs/repro_singleseq_base.yaml index 177e6613..bde1abb5 100644 --- a/projects/implicitron_trainer/configs/repro_singleseq_base.yaml +++ b/projects/implicitron_trainer/configs/repro_singleseq_base.yaml @@ -10,7 +10,6 @@ data_source_args: images_per_seq_options: - 2 dataset_map_provider_JsonIndexDatasetMapProvider_args: - dataset_name: singlesequence assert_single_seq: true n_frames_per_sequence: -1 test_restrict_sequence_id: 0