Graph Machine Learning
AnemoI
English
jpxkqx commited on
Commit
e827434
·
verified ·
1 Parent(s): e1220cb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -0
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