zhangrenchao commited on
Commit
001e867
·
verified ·
1 Parent(s): 01fa30d

Add English model card

Browse files
Files changed (1) hide show
  1. README.md +92 -0
README.md ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ tags:
6
+ - OneScience
7
+ - Earth Science
8
+ - Seasonal Prediction
9
+ - ACE2
10
+ - Ensemble Forecasting
11
+ - NAO
12
+ frameworks: PyTorch
13
+ ---
14
+
15
+ <p align="center"><strong><span style="font-size: 30px;">ACE2-Seasonal</span></strong></p>
16
+
17
+ # Model Introduction
18
+
19
+ ACE2-Seasonal reproduces the global seasonal hindcast experiment using ACE2-ERA5. Six-hour autoregression and persistent SST and sea-ice anomalies form a 64-member lagged ensemble.
20
+
21
+ Paper: Skilful global seasonal predictions from a machine learning weather model trained on reanalysis data
22
+ https://doi.org/10.1038/s41612-025-01198-3
23
+
24
+ # Model Description
25
+
26
+ The study was conducted by teams at the Met Office, University of Exeter, and Ai2. ACE2 was trained with ERA5 atmospheric reanalysis, while seasonal experiments use ERA5 initial states and persistent ocean-boundary anomalies. It supports global DJF prediction and NAO, ENSO-teleconnection, and ensemble spread-skill analysis.
27
+
28
+ # Use Cases
29
+
30
+ | Use Case | Description |
31
+ | :---: | :--- |
32
+ | Seasonal ensembles | Construct a 64-member lagged ensemble. |
33
+ | NAO prediction | Evaluate ensemble-mean NAO correlation and spread. |
34
+ | Boundary forcing | Persist initialized SST and sea-ice anomalies. |
35
+ | ModelScope/OneCode execution | Validate structured data, training, inference, seasonal metrics, and visualization. |
36
+ | Multi-GPU training | Start multi-process training through `torchrun`. |
37
+
38
+ # Usage Instructions
39
+
40
+ ```bash
41
+ hf download OneScience-Group/ACE2-Seasonal --local-dir ./ACE2-Seasonal
42
+ cd ACE2-Seasonal
43
+ ```
44
+
45
+ ### Environment Dependencies
46
+
47
+ **Hardware Requirements**
48
+
49
+ - A GPU or DCU is recommended.
50
+ - A CPU can be used for connectivity validation with the default small-sample configuration.
51
+ - DCU users should install DTK 25.04.2 or a compatible OneScience-recommended version first.
52
+
53
+ **DCU Environment**
54
+
55
+ ```bash
56
+ # Activate DTK and Conda first
57
+ conda create -n onescience311 python=3.11 -y
58
+ conda activate onescience311
59
+ pip install onescience[earth-dcu] -i http://mirrors.onescience.ai:3141/pypi/simple/ --trusted-host mirrors.onescience.ai
60
+ ```
61
+
62
+ **GPU Environment**
63
+
64
+ ```bash
65
+ # Activate Conda first
66
+ conda create -n onescience311 python=3.11 -y libstdcxx-ng=12 libgcc-ng=12 gcc_linux-64=12 gxx_linux-64=12
67
+ conda activate onescience311
68
+ pip install onescience[earth-gpu] -i http://mirrors.onescience.ai:3141/pypi/simple/ --trusted-host mirrors.onescience.ai
69
+ ```
70
+
71
+ ```bash
72
+ python scripts/fake_data.py
73
+ python scripts/train.py
74
+ torchrun --nproc_per_node=2 --nnodes=1 --master_addr="localhost" --master_port=29500 scripts/train.py
75
+ python scripts/inference.py
76
+ python scripts/result.py
77
+ ```
78
+
79
+ Synthetic data retain the global-grid protocol, boundary variables, six-hour order, and 64-member ensemble. The six-hour state-transition target completes backpropagation and parameter updates in verified single-process and two-process DDP runs. Inference restores the checkpoint, performs 12 engineering forecast steps, and produces a finite `[64,12,8,16,16]` ensemble while retaining the incomplete-coverage marker. Training results are saved to `result/checkpoints/ace2_seasonal.pt`; ensemble predictions and seasonal metrics are saved under `result/output/` and `result/evaluation/`.
80
+
81
+ # Official OneScience Information
82
+
83
+ | Platform | OneScience Main Repository | Skills Repository |
84
+ |---|---|---|
85
+ | Gitee | https://gitee.com/onescience-ai/onescience | https://gitee.com/onescience-ai/oneskills |
86
+ | GitHub | https://github.com/onescience-ai/OneScience | https://github.com/onescience-ai/oneskills |
87
+
88
+ # Citation and License
89
+
90
+ This repository is an independent engineering reproduction of the public ACE2-Seasonal specifications.
91
+
92
+ The original paper is licensed under CC BY 4.0; the paper, ACE2 code, weights, and ERA5 data remain subject to their respective licenses and terms.