Fix metric log name, added fuser config option

metno · Dec 6, 2024 · 752093d · 752093d
1 parent df83aaf
commit 752093d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/anemoi/training/config/model/fuser.yaml b/src/anemoi/training/config/model/fuser.yaml
@@ -1,5 +1,6 @@
 activation: GELU
 num_channels: 1024
+use_obs_fuser: True
 
 model:
   _target_: anemoi.models.models.obsfuser.AnemoiObsFuser

diff --git a/src/anemoi/training/train/netatmo_forecaster.py b/src/anemoi/training/train/netatmo_forecaster.py
@@ -319,7 +319,7 @@ def training_step(self, batch: list, batch_idx: int) -> torch.Tensor:
         )
         for dset, loss in enumerate(train_loss):
             self.log(
-                f"train_{getattr(self.loss.losses[dset], 'name', self.loss.losses[dset].__class__.__name__.lower())}_{dset}",
+                f"train_{getattr(self.loss.losses[dset], 'name', self.loss.losses[dset].__class__.__name__.lower())}_dset{dset}",
                 loss,
                 on_epoch=True,
                 on_step=True,