device: auto
output_dir: ../../../result/training/reference
seed: {value: 42, deterministic: false}
logging:
  wandb: {enabled: false}
dataset: {type: fast}
output_root: ../../../result/final_preprocess/training_full
catalog_path: ../../../result/final_preprocess/training_full/catalog.csv
input_window: {past_minutes: 50, interval_minutes: 10}
input_sources: [concat, hsr]
labels:
  bt: {future_minutes: 60, interval_minutes: 10}
train:
  split: train
  time_ranges:
    - {start: "201907010000", end: "201910312350"}
    - {start: "202005010000", end: "202010312350"}
    - {start: "202105010000", end: "202110312350"}
    - {start: "202205010000", end: "202210312350"}
    - {start: "202305010000", end: "202310312350"}
  input_sources: [concat, hsr]
  required_labels: [ci_hard, bt]
  epochs: 250
  batch_size: 3
  num_workers: 2
  persistent_workers: true
  prefetch_factor: 2
  shuffle: true
  drop_last: true
  grad_clip_norm: null
  use_ram_cache: false
  use_ram_chunk: true
  resume: true
  resume_path: null
valid:
  split: valid
  time_ranges:
    - {start: "202405010000", end: "202410312350"}
  input_sources: [concat, hsr]
  label_key: ci_hard
  required_labels: [ci_hard]
  batch_size: 2
  num_workers: 2
  shuffle: false
  drop_last: false
  use_ram_cache: true
  use_ram_chunk: false
ram_cache: {use_ram_cache: false}
ram_chunk:
  enabled: false
  chunk_ram_gb: 44.0
  async_prefetch: true
  read_block_rows: 64
  cache_dtype: float16
  swap_policy: repeat_current
  chunk_order: circular
  verbose: true
dataloader: {pin_memory: true}
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
loss:
  name: ci_bt
  params: {normalize_terms: true, ref_batches: 100, eps: 0.00000001}
  ci_loss:
    name: bce
    output_key: ci
    label_key: ci_hard
    weight: 0.85
    params: {eps: 0.000001}
  bt_loss:
    name: bt_mse
    output_key: bt
    label_key: bt
    weight: 0.15
    params: {}
optimizer: {name: adamw, params: {lr: 0.001, weight_decay: 0.0001}}
scheduler: {name: step, params: {step_size: 10, gamma: 0.9}}
validation:
  checkpoint_glob: null
  start_epoch: null
  checkpoint_stride: 2
  max_checkpoints: null
  output_dir: null
  eval_workers: 16
  thresholds: [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9]
  cluster_mode: mode_like
  min_cluster_pixels: 3
  label_min_cluster_pixels: 0
  merge_buffer_km: 12
  label_buffer_km: 0
  model_false_buffer_km: 0
  cluster_mask_expansion_km: 0
  pixel_size_km: 2.0
  connectivity: 8
  copy_best_model: true