device: auto
seed: {value: 42, deterministic: false}
dataset:
  type: fast
output_root: ../../../result/final_preprocess/2025
catalog_path: ../../../result/final_preprocess/2025/catalog.csv
input_window:
  past_minutes: 50
  interval_minutes: 10
input_sources: [concat, hsr]
labels:
  bt:
    future_minutes: 60
    interval_minutes: 10
    use_mask: false
model:
  name: simvp_bt_aux
  params:
    in_shape: [6, 12, 583, 550]
    out_shape: [1, 1, 583, 550]
    bt_out_shape: [6, 1, 583, 550]
    hid_S: 64
    hid_T: 256
    N_S: 6
    N_T: 8
    model_type: gSTA
    groups: 1
    mlp_ratio: 8.0
    drop: 0.1
    drop_path: 0.0
    spatio_kernel_enc: 3
    spatio_kernel_dec: 3
inference:
  checkpoint_path: ../../../result/training/checkpoints/model.safetensors
  output_dir: ../../../result/training/inference/2025
  time_ranges:
    - {start: "202505010000", end: "202510312350"}
  require_label: true
  label_key: ci_hard
  batch_size: 1
  num_workers: 0
  output_key: ci
  save_legacy_batch_dim: true
  dtype: float32
  save_summary: true
  skip_existing: true
dataloader: {pin_memory: true}
radar_mask:
  mode: masked
  source: hsr_valid_mask