defaults:
  - data: openpano-radial
  - train: geocalib
  - model: geocalib
  - _self_

data:
  # smaller batch size since lm takes more memory
  train_batch_size: 18
  val_batch_size: 18
  test_batch_size: 18

model:
  optimizer:
    camera_model: simple_radial

  weights: weights/geocalib.tar

train:
  lr: 1e-5 # smaller lr since we are fine-tuning
  num_steps: 200_000 # adapt to see same number of samples as previous training

  lr_schedule:
    type: SequentialLR
    on_epoch: false
    options:
      # adapt to see same number of samples as previous training
      milestones: [5_000]
      schedulers:
        - type: LinearLR
          options:
            start_factor: 1e-3
            total_iters: 5_000
        - type: MultiStepLR
          options:
            gamma: 0.1
            # adapt to see same number of samples as previous training
            milestones: [110_000, 170_000]