zhangrenchao commited on
Commit
4ece04d
·
verified ·
1 Parent(s): ae605ea

Add English model card

Browse files
Files changed (1) hide show
  1. README.md +56 -0
README.md ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ tags:
6
+ - OneScience
7
+ - Earth Science
8
+ - Data Assimilation
9
+ - Global Weather
10
+ - Satellite Observations
11
+ - Cascaded Forecasting
12
+ frameworks: PyTorch
13
+ ---
14
+ <p align="center"><strong><span style="font-size: 30px;">FuXi-Weather</span></strong></p>
15
+ # Model Introduction
16
+ FuXi-Weather maps raw satellite observations to global forecasts through FuXi-DA and cascaded FuXi forecast models.
17
+
18
+ Paper: A data-to-forecast machine learning system for global weather
19
+ https://doi.org/10.1038/s41467-025-62024-1
20
+ # Model Description
21
+ The system was proposed by teams from the Shanghai Academy of Artificial Intelligence for Science, Fudan University, CMA, and collaborators. It was trained with ERA5, microwave radiances from three polar-orbiting satellites, and GNSS radio occultation. Masked latent assimilation and Short/Medium forecasting support six-hour cycling and global forecasts to ten days.
22
+ # Use Cases
23
+ | Use Case | Description |
24
+ |---|---|
25
+ | Satellite assimilation | Fuse sparse observations and forecast backgrounds. |
26
+ | Global forecasting | Cascade short- and medium-range models. |
27
+ | Cycling analysis | Update global analyses and forecasts every six hours. |
28
+ | ModelScope/OneCode execution | Validate data, training, inference, metrics, and visualization. |
29
+ | Multi-GPU training | Start multi-process training through `torchrun`. |
30
+ # Usage Instructions
31
+ Use a GPU or DCU when available; CPU supports the default smoke configuration. DCU users should install a compatible DTK release.
32
+ ```bash
33
+ hf download OneScience-Group/FuXi-Weather --local-dir ./FuXi-Weather
34
+ cd FuXi-Weather
35
+ python scripts/fake_data.py
36
+ ```
37
+ For single-process training, use:
38
+ ```bash
39
+ python scripts/train.py
40
+ ```
41
+ For multi-process training, use:
42
+ ```bash
43
+ torchrun --standalone --nproc_per_node=2 scripts/train.py
44
+ ```
45
+ Run inference and evaluation with:
46
+ ```bash
47
+ python scripts/inference.py
48
+ python scripts/result.py
49
+ ```
50
+ Training jointly optimizes analysis and forecast objectives. Inference produces finite `[2,12,20,16,16]` cascaded forecasts, while evaluation reports lead-time RMSE under `result/evaluation/`.
51
+ ## Trained Weights
52
+ No weights are bundled under `weight/`. The FuXi model is available at https://zenodo.org/records/10401602, and the FuXi Weather model used by the paper is available at https://zenodo.org/records/15762985.
53
+ # Citation and License
54
+ This repository is an independent engineering reproduction of the public FuXi-Weather specifications.
55
+
56
+ The original paper is licensed under CC BY-NC-ND 4.0; the original paper, official code, model weights, and related data remain subject to their respective licenses and terms.