anaprietonem
commited on
Commit
•
cba8564
1
Parent(s):
29a9975
Update README.md
Browse files
README.md
CHANGED
@@ -80,7 +80,11 @@ ai-models anemoi --checkpoint aifs_single_v0.2.1.ckpt --file example_20241107_12
|
|
80 |
|
81 |
**Note** we train AIFS using `flash_attention` (https://github.com/Dao-AILab/flash-attention).
|
82 |
There are currently some issues when trying to install flash attention with the latest PyTorch version 2.5 and CUDA 12.4 (https://github.com/Dao-AILab/flash-attention/issues/1330).
|
83 |
-
For that reason, we recommend you install PyTorch 2.4.
|
|
|
|
|
|
|
|
|
84 |
|
85 |
After running the `ai-models` command the output of the forecast should be written into `anemoi.grib`
|
86 |
Below you can find an example to read that file and load it as numpy array or xarray.
|
@@ -178,7 +182,7 @@ sdaf) can be found in e.g Ben Bouallegue et al. ` [2024].
|
|
178 |
</div>
|
179 |
|
180 |
|
181 |
-
Forecasts are initialised on 00 and 12 UTC.
|
182 |
southern extra-tropics (s.hem), tropics and Europe. Blue colours mark score improvements and red colours score
|
183 |
degradations. Purple colours indicate an increased in standard deviation of forecast anomaly, while green colours
|
184 |
indicate a reduction. Framed rectangles indicate 95% significance level. Variables are geopotential (z), temperature
|
@@ -188,6 +192,13 @@ suffix indicates verification against IFS NWP analyses (an) or radiosonde and SY
|
|
188 |
shown are anomaly correlation (ccaf), SEEPS (seeps, for precipitation), RMSE (rmsef) and standard deviation of
|
189 |
forecast anomaly (sdaf, see text for more explanation).
|
190 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
191 |
## Technical Specifications
|
192 |
|
193 |
### Hardware
|
|
|
80 |
|
81 |
**Note** we train AIFS using `flash_attention` (https://github.com/Dao-AILab/flash-attention).
|
82 |
There are currently some issues when trying to install flash attention with the latest PyTorch version 2.5 and CUDA 12.4 (https://github.com/Dao-AILab/flash-attention/issues/1330).
|
83 |
+
For that reason, we recommend you install PyTorch 2.4.
|
84 |
+
Additonally the use of 'Flash Attention' package also imposes certain requirements in terms of software and hardware. Those can be found under #Installation and Features in https://github.com/Dao-AILab/flash-attention
|
85 |
+
|
86 |
+
|
87 |
+
|
88 |
|
89 |
After running the `ai-models` command the output of the forecast should be written into `anemoi.grib`
|
90 |
Below you can find an example to read that file and load it as numpy array or xarray.
|
|
|
182 |
</div>
|
183 |
|
184 |
|
185 |
+
Forecasts are initialised on 00 and 12 UTC. The scorecard show relative score changes as function of lead time (day 1 to 10) for northern extra-tropics (n.hem),
|
186 |
southern extra-tropics (s.hem), tropics and Europe. Blue colours mark score improvements and red colours score
|
187 |
degradations. Purple colours indicate an increased in standard deviation of forecast anomaly, while green colours
|
188 |
indicate a reduction. Framed rectangles indicate 95% significance level. Variables are geopotential (z), temperature
|
|
|
192 |
shown are anomaly correlation (ccaf), SEEPS (seeps, for precipitation), RMSE (rmsef) and standard deviation of
|
193 |
forecast anomaly (sdaf, see text for more explanation).
|
194 |
|
195 |
+
Additional evaluation analysis including tropycal cyclone performance or comparison against other popular data-driven models can be found in AIFS preprint (https://arxiv.org/pdf/2406.01465v1) section 4.
|
196 |
+
|
197 |
+
# Known limitations
|
198 |
+
- This version of AIFS shares certain limitations with some of the other data-driven weather forecast models that are trained with a weighted MSE loss, such as blurring of the forecast fields at longer lead times.
|
199 |
+
- AIFS exhibits reduced forecast skill in the stratosphere forecast owing to the linear loss scaling with height
|
200 |
+
- AIFS currently provides reduced intensity of some high-impact systems such as tropical cyclones.
|
201 |
+
|
202 |
## Technical Specifications
|
203 |
|
204 |
### Hardware
|