model:
  in_channels: 12
  patch_size: 8
  spectral_patch_size: 3
  embed_dim: 32
  encoder_depth: 1
  encoder_heads: 4
  decoder_dim: 32
  decoder_depth: 1
  decoder_heads: 4
  mask_ratio: 0.90
  spectral_angle_weight: 0.1
  spectral_gradient_weight: 0.1
stages:
  - name: stage1
    dataset: fMoW-Sentinel
    image_size: 96
    train_path: ./data/stage1_train.npz
    train_samples: 2
    epochs: 1
  - name: stage2
    dataset: BigEarthNet
    image_size: 128
    train_path: ./data/stage2_train.npz
    train_samples: 2
    epochs: 1
data:
  test_path: ./data/stage2_test.npz
  test_samples: 1
  protocol: spectralgpt_progressive_s2_v2
training:
  batch_size: 1
  learning_rate: 0.0001
  weight_decay: 0.05
  save_dir: ./result/checkpoints
  checkpoint: ./result/checkpoints/final.pth
  metrics: ./result/training/metrics.json
  amp: true
runtime:
  device: cpu
  seed: 42
  output_dir: ./result/output