Climate2Weather / conf /config.yaml
zhangrenchao's picture
Publish Climate2Weather reproduction
21b9cf1 verified
Raw History Blame Contribute Delete
644 Bytes
seed: 42
data: {path: data/downscale.npz, samples: 6, variables: 4, window: 3, coarse_grid: [8, 8], fine_grid: [128, 128]}
model: {channels: 12, hidden: 16}
train: {epochs: 1, learning_rate: 0.001}
inference: {members: 8, denoise_steps: 4}
paths: {checkpoint: result/checkpoints/climate2weather.pt, training_metrics: result/training/metrics.json, predictions: result/output/predictions.npz, evaluation: result/evaluation/metrics.json, figure: result/evaluation/comparison.png}
paper_model: {variables: [uas, vas, tas, psl], coarse_grid: [8, 8], fine_grid: [128, 128], coarse_hours: 6, fine_hours: 1, train_years: [2006, 2013], test_year: 2014}