| 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 |
|
|
