seed: 42
data:
  path: data/corrdiff.npz
  protocol: corrdiff_npz_v1
  input_key: input
  target_key: target
  input_shape: [12, 36, 36]
  target_shape: [4, 448, 448]
  input_variables: [tcwv, t2m, u10m, v10m, t500, z500, u500, v500, t850, z850, u850, v850]
  target_variables: [t2m, u10m, v10m, maximum_radar_reflectivity]
  fake_samples: 2
model:
  in_channels: 12
  out_channels: 4
  base_channels: 8
  feature_size: 28
  output_size: 448
  sigma_data: 0.5
training:
  regression_epochs: 1
  diffusion_epochs: 1
  batch_size: 1
  max_batches_per_epoch: 1
  learning_rate: 0.0002
  num_workers: 0
  amp: true
  p_mean: -1.2
  p_std: 1.2
sampling:
  batch_size: 1
  ensemble_size: 2
  steps: 3
  sigma_min: 0.002
  sigma_max: 5.0
  rho: 7.0
  solver: heun
runtime:
  device: auto
paths:
  checkpoint: result/checkpoints/corrdiff.pt
  training_metrics: result/training/metrics.json
  predictions: result/output/predictions.npz
  evaluation_dir: result/evaluation