Update README.md
Browse files
README.md
CHANGED
|
@@ -134,6 +134,22 @@ The use of 'Flash Attention' package also imposes certain requirements in terms
|
|
| 134 |
🚨 **Note** the `aifs_single_v1.0.ckpt` checkpoint just contains the model’s weights.
|
| 135 |
That file does not contain any information about the optimizer states, lr-scheduler states, etc.
|
| 136 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 137 |
|
| 138 |
## Training Details
|
| 139 |
|
|
|
|
| 134 |
🚨 **Note** the `aifs_single_v1.0.ckpt` checkpoint just contains the model’s weights.
|
| 135 |
That file does not contain any information about the optimizer states, lr-scheduler states, etc.
|
| 136 |
|
| 137 |
+
## How to train AIFS Single v1.0
|
| 138 |
+
|
| 139 |
+
To train this model you can use the configuration files included in this repository and the following Anemoi packages:
|
| 140 |
+
|
| 141 |
+
```
|
| 142 |
+
anemoi-training==0.3.1
|
| 143 |
+
anemoi-models==0.4.0
|
| 144 |
+
anemoi-graphs>=0.4.4
|
| 145 |
+
```
|
| 146 |
+
and run:
|
| 147 |
+
|
| 148 |
+
```
|
| 149 |
+
export DATASETS_PATH=???????
|
| 150 |
+
export OUTPUT_PATH=???????
|
| 151 |
+
anemoi-training train --config-name=config_pretraining.yaml
|
| 152 |
+
```
|
| 153 |
|
| 154 |
## Training Details
|
| 155 |
|