zhangrenchao commited on
Commit
699b17a
·
verified ·
1 Parent(s): 21b9cf1

Add English model card

Browse files
Files changed (1) hide show
  1. README.md +57 -0
README.md ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ tags:
6
+ - OneScience
7
+ - Earth Science
8
+ - Climate Downscaling
9
+ - Diffusion
10
+ - Probabilistic Forecasting
11
+ - Coherence
12
+ frameworks: PyTorch
13
+ ---
14
+ <p align="center"><strong><span style="font-size: 30px;">Climate2Weather</span></strong></p>
15
+ # Model Introduction
16
+ Climate2Weather uses conditional score diffusion to transform coarse climate simulations into probabilistic high-resolution weather trajectories.
17
+
18
+ Paper: A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulation
19
+ https://doi.org/10.1038/s41612-025-01157-y
20
+ # Model Description
21
+ The method was proposed by the University of Tübingen and Tübingen AI Center. It was trained with 2006–2013 COSMO-REA6 reanalysis and conditions on climate-model fields only during inference. Score-based data assimilation jointly downscales four variables from `8x8` to `128x128` and from six-hourly to hourly resolution.
22
+ # Use Cases
23
+ | Use Case | Description |
24
+ |---|---|
25
+ | Probabilistic downscaling | Generate high-resolution ensemble trajectories. |
26
+ | Coherent generation | Jointly model variables and time. |
27
+ | Multivariate generation | Downscale wind, temperature, and sea-level pressure jointly. |
28
+ | Climate impacts | Generate fine-scale drivers for regional impact studies. |
29
+ | ModelScope/OneCode execution | Validate data, training, inference, metrics, and visualization. |
30
+ | Multi-GPU training | Start multi-process training through `torchrun`. |
31
+ # Usage Instructions
32
+ Use a GPU or DCU when available; CPU supports the default smoke configuration.
33
+ ```bash
34
+ hf download OneScience-Group/Climate2Weather --local-dir ./Climate2Weather
35
+ cd Climate2Weather
36
+ python scripts/fake_data.py
37
+ ```
38
+ For single-process training, use:
39
+ ```bash
40
+ python scripts/train.py
41
+ ```
42
+ For multi-process training, use:
43
+ ```bash
44
+ torchrun --standalone --nproc_per_node=2 scripts/train.py
45
+ ```
46
+ Run inference and evaluation with:
47
+ ```bash
48
+ python scripts/inference.py
49
+ python scripts/result.py
50
+ ```
51
+ Training minimizes denoising score matching. Inference produces a finite `[8,3,4,128,128]` ensemble with positive spread; evaluation reports RMSE, spread, PIT, and temporal differences.
52
+ ## Trained Weights
53
+ No weights are bundled under `weight/`. The authors provide trained diffusion-model weights and experiment code at https://github.com/schmidtjonathan/Climate2Weather; this compact implementation does not claim compatibility.
54
+ # Citation and License
55
+ This repository is an independent engineering reproduction of the public Climate2Weather specifications.
56
+
57
+ The original paper is licensed under CC BY 4.0; the original paper, official code, model weights, and related data remain subject to their respective licenses and terms.