seed: 42
device: auto

data:
  mode: synthetic
  path: ./data/rainnet_fake.hdf5
  input_steps: 4
  forecast_steps: 1
  interval_minutes: 5
  raw_height: 900
  raw_width: 900
  padded_height: 928
  padded_width: 928
  pixel_size_km: 1
  train_frames: 8
  val_frames: 6
  test_frames: 16
  num_workers: 0

model:
  in_channels: 4
  out_channels: 1
  dropout: 0.5
  mode: regression

train:
  epochs: 1
  batch_size: 1
  learning_rate: 0.0001
  max_train_batches: 1
  max_valid_batches: 1
  checkpoint_last: ./weight/rainnet_last.pth
  checkpoint_best: ./weight/rainnet_best.pth

paper_reference:
  epochs: 10
  batch_size: 2
  learning_rate: 0.0001
  raw_size: 900
  padded_size: 928

inference:
  checkpoint: ./weight/rainnet_best.pth
  rollout_steps: 12
  output_dir: ./result/output

evaluation:
  thresholds_mm_h:
    - 0.125
    - 1.0
    - 5.0
    - 10.0
    - 15.0
  fss_windows_km:
    - 1
    - 5
    - 10
    - 20
  result_dir: ./result