From 8797fb3cc092378b1907e47cd9a8dc4b003c3a5f Mon Sep 17 00:00:00 2001 From: icedoom888 Date: Wed, 18 Dec 2024 19:28:02 +0100 Subject: [PATCH] Rebased on develop --- src/anemoi/training/train/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/anemoi/training/train/train.py b/src/anemoi/training/train/train.py index f6e7f895..d92a9cc3 100644 --- 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