[bugfix] loading only the weights of the checkpoint

MeteoSwiss · Dec 10, 2024 · 2df18a7 · 2df18a7
1 parent da26cb7
commit 2df18a7
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
@@ -162,7 +162,7 @@ def model(self) -> GraphForecaster:
 
             LOGGER.info("Restoring only model weights from %s", self.last_checkpoint)
 
-            return model.load_from_checkpoint(self.last_checkpoint, **kwargs, strict=False)
+            return GraphForecaster.load_from_checkpoint(self.last_checkpoint, **kwargs, strict=False)
 
         LOGGER.info("Model initialised from scratch.")
         return model