WeatherBench2 / conf /config.yaml
zhangrenchao's picture
Publish WeatherBench2 reproduction
f58cc34 verified
Raw History Blame Contribute Delete
367 Bytes
seed: 42
data: {path: data/data.npz, samples: 8, variables: 8, grid: [24, 48]}
model: {variables: 8, hidden: 16}
train: {lr: 0.001}
paths: {checkpoint: result/checkpoints/weatherbench2.pt, training_metrics: result/training/metrics.json, predictions: result/output/predictions.npz, evaluation: result/evaluation/metrics.json, figure: result/evaluation/comparison.png}