project:
  name: SEEDS_develop
  model_name: SEEDS
  task: generative_ensemble_emulation
  seed: 42

paths:
  data_dir: data
  checkpoint_dir: data/checkpoint
  result_dir: result
  train_data: data/train.npz
  val_data: data/val.npz
  test_data: data/test.npz
  checkpoint: data/checkpoint/model_bak.pth

data:
  variables:
    - mean_sea_level_pressure
    - temperature_2m
    - eastward_wind_850hpa
    - northward_wind_850hpa
    - geopotential_500hpa
    - temperature_850hpa
    - total_column_water_vapour
    - specific_humidity_500hpa
  faces: 6
  height: 48
  width: 48
  seed_count: 2
  target_count: 16
  lead_time_days: 7
  dtype: float32

model:
  patch_size: 12
  embed_dim: 768
  spatial_layers: 6
  field_layers: 4
  sequence_layers: 6
  mlp_ratio: 4
  dropout: 0.0
  sigma_min: 0.01
  sigma_max: 100.0
  diffusion_time_max: 1.0
  prediction: score_noise

training:
  batch_size: 1
  gradient_accumulation_steps: 8
  epochs: 10
  learning_rate: 0.0001
  weight_decay: 0.0
  grad_clip_norm: 1.0
  num_workers: 0
  device: auto
  resume: null
  finetune: false

sampling:
  steps: 16
  data_batch_size: 1
  member_batch_size: 1
  solver: euler_maruyama

validation:
  max_batches: null
  plot_variable: temperature_2m