| seed: 42 | |
| data: {path: data/aerosol.npz, samples: 8, channels: 54, tile: 16, logical_grid: [361, 576]} | |
| model: {channels: 54, hidden: 24, heads: 4} | |
| train: {epochs: 1, learning_rate: 0.001, leads: [3, 6, 9, 12]} | |
| inference: {hours: 120} | |
| paths: {checkpoint: result/checkpoints/ai_gamfs.pt, training_metrics: result/training/metrics.json, predictions: result/output/predictions.npz, evaluation: result/evaluation/metrics.json, figure: result/evaluation/comparison.png} | |
| paper_model: {grid: [361, 576], channels: 54, aerosol_variables: 12, upper_levels: 9, leads: [3, 6, 9, 12], epochs: 80, forecast_hours: 120, paper_license: CC-BY-NC-ND-4.0} | |