seed: 42
data:
  root: data
  protocol: synthetic_SDO_AIA_HMI
  image_size: 32
  channels: 13
  input_steps: 2
  forecast_steps: 4
  train_samples: 32
  test_samples: 8
  channel_names: [AIA_94, AIA_131, AIA_171, AIA_193, AIA_211, AIA_304, AIA_335, AIA_1600, HMI_magnetogram, HMI_continuum, HMI_doppler, HMI_vector_x, HMI_vector_y]
  channel_mean: [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]
  channel_std: [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
model:
  image_size: 32
  patch_size: 4
  channels: 13
  input_steps: 2
  embed_dim: 64
  depth: 4
  spectral_blocks: 1
  num_heads: 4
  window_size: 3
  global_tokens: 4
training:
  epochs: 2
  one_step_epochs: 1
  rollout_epochs: 1
  batch_size: 4
  learning_rate: 0.0005
  weight_decay: 0.01
  num_workers: 0
  accum_iter: 1
  warmup_epochs: 1
  min_learning_rate: 0.00001
  amp: true
  grad_clip: 1.0
runtime:
  device: auto
  amp: true
paths:
  checkpoint: result/checkpoints/surya.pt
  training_metrics: result/training/metrics.json
  inference_dir: result/output
  evaluation_dir: result/evaluation