| seed: 42 | |
| data: | |
| format_version: weatherbench_v1 | |
| path: data/weatherbench.npz | |
| grid: [32, 64] | |
| samples: 24 | |
| channels: [Z500, T850] | |
| model: | |
| hidden_channels: 16 | |
| layers: 5 | |
| kernel_size: 5 | |
| train: | |
| epochs: 2 | |
| learning_rate: 0.001 | |
| paths: | |
| checkpoint: result/checkpoints/weatherbench_cnn.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: [32, 64] | |
| input_channels: [Z500, T850] | |
| conv_layers: 5 | |
| hidden_channels: 64 | |
| kernel_size: 5 | |
| activation: ELU | |
| optimizer: Adam | |
| loss: MSE | |
| test_years: [2017, 2018] | |