FuXi-Weather / conf /config.yaml
zhangrenchao's picture
Publish FuXi-Weather reproduction
ae605ea verified
Raw History Blame Contribute Delete
609 Bytes
seed: 42
data: {path: data/fuxi_weather.npz, samples: 8, channels: 20, tile: 16, logical_grid: [721, 1440]}
model: {channels: 20, hidden: 24}
train: {epochs: 1, learning_rate: 0.001}
inference: {steps: 12, paper_steps: 40, switch_step: 6}
paths: {checkpoint: result/checkpoints/fuxi_weather.pt, training_metrics: result/training/metrics.json, predictions: result/output/predictions.npz, evaluation: result/evaluation/metrics.json, figure: result/evaluation/comparison.png}
paper_model: {resolution_degrees: 0.25, cycle_hours: 6, observation_window_hours: 8, forecast_days: 10, paper_license: CC-BY-NC-ND-4.0}