"pcl_rl/controller.py" did not exist on "6f38af5fe714d434626921786f7ba706584167ce"
repro_multiseq_base.yaml 921 Bytes
Newer Older
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
1
2
3
defaults:
- repro_base.yaml
- _self_
4
data_source_ImplicitronDataSource_args:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
5
  data_loader_map_provider_SequenceDataLoaderMapProvider_args:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
6
    batch_size: 10
7
8
    dataset_length_train: 1000
    dataset_length_val: 1
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
9
    num_workers: 8
10
11
12
    train_conditioning_type: SAME
    val_conditioning_type: SAME
    test_conditioning_type: SAME
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
13
14
15
16
17
18
19
20
21
22
    images_per_seq_options:
    - 2
    - 3
    - 4
    - 5
    - 6
    - 7
    - 8
    - 9
    - 10
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
23
  dataset_map_provider_JsonIndexDatasetMapProvider_args:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
24
    assert_single_seq: false
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
25
    task_str: multisequence
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
26
27
28
    n_frames_per_sequence: -1
    test_on_train: true
    test_restrict_sequence_id: 0
29
30
optimizer_factory_ImplicitronOptimizerFactory_args:
  multistep_lr_milestones:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
31
  - 1000
32
33
34
35
36
37
training_loop_ImplicitronTrainingLoop_args:
  max_epochs: 3000
  evaluator_ImplicitronEvaluator_args:
    camera_difficulty_bin_breaks:
      - 0.666667
      - 0.833334
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
38
    is_multisequence: true