feature: make models instantiateable

ecmwf · Sep 5, 2024 · 5393803 · 5393803
1 parent cf0e594
commit 5393803
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/src/anemoi/training/config/model/gnn.yaml b/src/anemoi/training/config/model/gnn.yaml
@@ -1,6 +1,9 @@
 activation: GELU
 num_channels: 512
 
+model:
+  _target_: anemoi.models.models.encoder_processor_decoder.AnemoiModelEncProcDec
+
 processor:
   _target_: anemoi.models.layers.processor.GNNProcessor
   _convert_: all

diff --git a/src/anemoi/training/config/model/graphtransformer.yaml b/src/anemoi/training/config/model/graphtransformer.yaml
@@ -1,6 +1,9 @@
 activation: GELU
 num_channels: 1024
 
+model:
+  _target_: anemoi.models.models.encoder_processor_decoder.AnemoiModelEncProcDec
+
 processor:
   _target_: anemoi.models.layers.processor.GraphTransformerProcessor
   _convert_: all

diff --git a/src/anemoi/training/config/model/transformer.yaml b/src/anemoi/training/config/model/transformer.yaml
@@ -1,6 +1,9 @@
 activation: GELU
 num_channels: 1024
 
+model:
+  _target_: anemoi.models.models.encoder_processor_decoder.AnemoiModelEncProcDec
+
 processor:
   _target_: anemoi.models.layers.processor.TransformerProcessor
   _convert_: all