zhangrenchao commited on
Commit
b81c600
·
verified ·
1 Parent(s): aa529c9

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
+ - WeatherBench
9
+ - Medium-Range Forecasting
10
+ - ERA5
11
+ - Benchmarking
12
+ frameworks: PyTorch
13
+ ---
14
+
15
+ <p align="center"><strong><span style="font-size: 30px;">WeatherBench</span></strong></p>
16
+
17
+ # Model Introduction
18
+
19
+ WeatherBench is a dataset and evaluation benchmark for data-driven global weather forecasting. This reproduction implements the paper's five-layer fully convolutional CNN for three- and five-day Z500 and T850 evaluation.
20
+
21
+ Paper: WeatherBench: A Benchmark Data Set for Data-Driven Weather Forecasting
22
+ https://arxiv.org/abs/2002.00469
23
+
24
+ # Model Description
25
+
26
+ The benchmark was proposed by teams from the Technical University of Munich, ECMWF, Stockholm University, the University of Washington, and the University of Toronto. It uses ERA5 reanalysis from 1979–2018 with multiresolution global fields and 13 pressure levels. The paper CNN predicts Z500 and T850 directly or iteratively and is compared with persistence, climatology, linear regression, and IFS.
27
+
28
+ # Use Cases
29
+
30
+ | Use Case | Description |
31
+ | :---: | :--- |
32
+ | Medium-range forecasting | Evaluate three- and five-day Z500 and T850. |
33
+ | Baseline comparison | Compare forecasts with persistence and climatology. |
34
+ | Latitude-weighted scoring | Compute WeatherBench RMSE and ACC. |
35
+ | ModelScope/OneCode execution | Validate structured data, training, inference, weather 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/WeatherBench --local-dir ./WeatherBench
42
+ cd WeatherBench
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 Z500 and T850 on the paper's `32x64` grid and the six-hour iterative protocol. The five-layer fully convolutional CNN completes MSE optimization and parameter updates in verified single-process and two-process DDP runs. Inference restores the checkpoint, performs 12-step and 20-step rollouts with periodic longitude boundaries, and produces finite outputs with shape `[4,2,2,32,64]`. Training results are saved to `result/checkpoints/weatherbench_cnn.pt`; forecasts and latitude-weighted 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 WeatherBench specifications.
91
+
92
+ The original paper, WeatherBench code, and ERA5 data remain subject to their respective licenses and terms.