Rebased on develop

MeteoSwiss · Dec 18, 2024 · 8797fb3 · 8797fb3
1 parent 498a792
commit 8797fb3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/anemoi/training/train/train.py b/src/anemoi/training/train/train.py
@@ -165,7 +165,7 @@ def model(self) -> GraphForecaster:
                     model = transfer_learning_loading(model, self.last_checkpoint)
             else:
                 LOGGER.info("Restoring only model weights from %s", self.last_checkpoint)
-                model = model.load_from_checkpoint(self.last_checkpoint, **kwargs, strict=False)
+                model = GraphForecaster.load_from_checkpoint(self.last_checkpoint, **kwargs, strict=False)
 
         if hasattr(self.config.training, "submodules_to_freeze"):
             # Freeze the chosen model weights