File size: 1,101 Bytes
a13f4b9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | 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
|