repro_multiseq_base.yaml 633 Bytes
Newer Older
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
1
2
3
defaults:
- repro_base.yaml
- _self_
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
4
data_source_args:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
5
  data_loader_map_provider_SequenceDataLoaderMapProvider_args:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
6
7
8
9
10
11
12
13
14
15
16
17
18
19
    batch_size: 10
    dataset_len: 1000
    dataset_len_val: 1
    num_workers: 8
    images_per_seq_options:
    - 2
    - 3
    - 4
    - 5
    - 6
    - 7
    - 8
    - 9
    - 10
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
20
  dataset_map_provider_JsonIndexDatasetMapProvider_args:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
21
    assert_single_seq: false
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
22
    task_str: multisequence
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
23
24
25
26
27
28
    load_point_clouds: false
    mask_depths: false
    mask_images: false
    n_frames_per_sequence: -1
    test_on_train: true
    test_restrict_sequence_id: 0
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
29
30
31
32
solver_args:
  max_epochs: 3000
  milestones:
  - 1000