project:
  name: stormcast
  seed: 42
  output_dir: ./outputs

data:
  root_dir: ./data
  train_years: [2019]
  val_years: [2021]
  test_years: [2022]
  time_step_hours: 1
  input_steps: 1
  output_steps: 1
  era5_image_size: [721, 1440]
  image_size: [512, 640]
  num_timesteps: 4
  normalize: true
  era5_variables:
    - u10m
    - v10m
    - t2m
    - tcwv
    - sp
    - msl
    - u1000
    - u850
    - u500
    - u250
    - v1000
    - v850
    - v500
    - v250
    - z1000
    - z850
    - z500
    - z250
    - t1000
    - t850
    - t500
    - t250
    - q1000
    - q850
    - q500
    - q250
  state_variables:
    - u10m
    - v10m
    - t2m
    - msl
    - u1hl
    - u2hl
    - u3hl
    - u4hl
    - u5hl
    - u6hl
    - u7hl
    - u8hl
    - u9hl
    - u10hl
    - u11hl
    - u13hl
    - u15hl
    - u20hl
    - u25hl
    - u30hl
    - v1hl
    - v2hl
    - v3hl
    - v4hl
    - v5hl
    - v6hl
    - v7hl
    - v8hl
    - v9hl
    - v10hl
    - v11hl
    - v13hl
    - v15hl
    - v20hl
    - v25hl
    - v30hl
    - t1hl
    - t2hl
    - t3hl
    - t4hl
    - t5hl
    - t6hl
    - t7hl
    - t8hl
    - t9hl
    - t10hl
    - t11hl
    - t13hl
    - t15hl
    - t20hl
    - t25hl
    - t30hl
    - q1hl
    - q2hl
    - q3hl
    - q4hl
    - q5hl
    - q6hl
    - q7hl
    - q8hl
    - q9hl
    - q10hl
    - q11hl
    - q13hl
    - q15hl
    - q20hl
    - q25hl
    - q30hl
    - Z1hl
    - Z2hl
    - Z3hl
    - Z4hl
    - Z5hl
    - Z6hl
    - Z7hl
    - Z8hl
    - Z9hl
    - Z10hl
    - Z11hl
    - Z13hl
    - Z15hl
    - Z20hl
    - Z25hl
    - Z30hl
    - p1hl
    - p2hl
    - p3hl
    - p4hl
    - p5hl
    - p6hl
    - p7hl
    - p8hl
    - p9hl
    - p10hl
    - p11hl
    - p13hl
    - p15hl
    - p20hl
    - refc
  invariant_variables: [lsm, orography]

dataloader:
  batch_size: 1
  num_workers: 0
  pin_memory: true

model:
  regression_conditions: [state, background, invariant]
  diffusion_conditions: [state, regression, invariant]
  image_size: [512, 640]
  model_channels: 128
  channel_mult: [1, 2, 2, 2, 2]
  num_blocks: 4
  attention_resolutions: []
  regression_weights: ./data/checkpoint/regression/model_bak.pt
  diffusion_weights: ./data/checkpoint/diffusion/model_bak.pt

training:
  stage: regression
  from_scratch: true
  initial_weights: null
  resume_checkpoint: null
  regression_weights: ./data/checkpoint/regression/model_bak.pt
  epochs: 10
  max_steps: 10
  learning_rate: 0.0002
  betas: [0.9, 0.999]
  weight_decay: 0.0
  gradient_clip_norm: null
  checkpoint_interval: 1
  log_interval: 1
  sigma_data: 0.5
  P_mean: -1.2
  P_std: 1.2
  checkpoint_dir: ./data/checkpoint

inference:
  split: test
  num_steps: 1
  diffusion_steps: 18
  sigma_min: 0.002
  sigma_max: 800.0
  rho: 7.0
  output_dir: ./outputs/inference
  plot_state_variable: refc
  plot_background_variable: t2m