zhangrenchao commited on
Commit
685c2e0
·
verified ·
1 Parent(s): bec9cbb

Add English model card

Browse files
Files changed (1) hide show
  1. README.md +86 -0
README.md ADDED
@@ -0,0 +1,86 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ tags:
6
+ - OneScience
7
+ - Earth Science
8
+ - Earth System Model
9
+ - Coupled Climate
10
+ - Long Simulation
11
+ - U-Net
12
+ frameworks: PyTorch
13
+ ---
14
+
15
+ <p align="center"><strong><span style="font-size: 30px;">DLESyM</span></strong></p>
16
+
17
+ # Model Introduction
18
+
19
+ DLESyM asynchronously couples deep-learning atmosphere and ocean modules for long free-running climate simulations and diagnostic precipitation.
20
+
21
+ Paper: A Deep Learning Earth System Model for Efficient Simulation of the Observed Climate
22
+ https://arxiv.org/abs/2409.16247
23
+
24
+ # Model Description
25
+
26
+ The model was proposed by an atmospheric-science and machine-learning research team. It was trained with 1983–2017 ERA5 fields, ISCCP OLR, and SST. Coupled DLWP, DLOM, and precipitation modules support current-climate simulation and internal-variability analysis.
27
+
28
+ # Use Cases
29
+
30
+ | Use Case | Description |
31
+ | :---: | :--- |
32
+ | Long climate simulation | Run stable atmosphere-ocean rollouts. |
33
+ | Climate variability | Analyze ENSO, monsoons, and annular modes. |
34
+ | Precipitation diagnosis | Diagnose accumulated precipitation from atmospheric states. |
35
+ | ModelScope/OneCode execution | Validate data, training, inference, climate metrics, and visualization. |
36
+ | Multi-GPU training | Start multi-process training through `torchrun`. |
37
+
38
+ # Usage Instructions
39
+
40
+ Use a GPU or DCU when available; CPU supports the default smoke configuration.
41
+
42
+ ```bash
43
+ hf download OneScience-Group/DLESyM --local-dir ./DLESyM
44
+ cd DLESyM
45
+ ```
46
+
47
+ ### Environment Dependencies
48
+
49
+ **Hardware Requirements**
50
+
51
+ - A GPU or DCU is recommended.
52
+ - A CPU can be used for connectivity validation with the default small-sample configuration.
53
+ - DCU users should install DTK 25.04.2 or a compatible OneScience-recommended version first.
54
+
55
+ **DCU Environment**
56
+
57
+ ```bash
58
+ # Activate DTK and Conda first
59
+ conda create -n onescience311 python=3.11 -y
60
+ conda activate onescience311
61
+ pip install onescience[earth-dcu] -i http://mirrors.onescience.ai:3141/pypi/simple/ --trusted-host mirrors.onescience.ai
62
+ ```
63
+
64
+ **GPU Environment**
65
+
66
+ ```bash
67
+ # Activate Conda first
68
+ conda create -n onescience311 python=3.11 -y libstdcxx-ng=12 libgcc-ng=12 gcc_linux-64=12 gxx_linux-64=12
69
+ conda activate onescience311
70
+ pip install onescience[earth-gpu] -i http://mirrors.onescience.ai:3141/pypi/simple/ --trusted-host mirrors.onescience.ai
71
+ ```
72
+
73
+ ```bash
74
+ python scripts/fake_data.py
75
+ python scripts/train.py
76
+ torchrun --standalone --nproc_per_node=2 scripts/train.py
77
+ python scripts/inference.py
78
+ python scripts/result.py
79
+ ```
80
+ Training jointly optimizes atmosphere, ocean, and precipitation modules. Inference runs four coupled cycles and evaluation reports finite drift diagnostics.
81
+ ## Trained Weights
82
+ No weights are bundled under `weight/`. The authors provide configurations and weights at https://github.com/AtmosSci-DLESM/DLESyM.
83
+ # Citation and License
84
+ This repository is an independent engineering reproduction of the public DLESyM specifications.
85
+
86
+ The original preprint, official code, model weights, and related data remain subject to their respective licenses and terms.