diff --git a/aurora/model/aurora.py b/aurora/model/aurora.py index 9fae270..4c68c6b 100644 --- a/aurora/model/aurora.py +++ b/aurora/model/aurora.py @@ -158,9 +158,6 @@ def forward(self, batch: Batch) -> Batch: Args: batch (:class:`Batch`): Batch to run the model on. - Raises: - ValueError: If no metric is provided. - Returns: :class:`Batch`: Prediction for the batch. """ diff --git a/docs/finetuning.md b/docs/finetuning.md index 8bdda7b..266f894 100644 --- a/docs/finetuning.md +++ b/docs/finetuning.md @@ -53,7 +53,7 @@ model = Aurora( static_vars=("lsm", "z", "slt", "new_static_var"), atmos_vars=("z", "u", "v", "t", "q", "new_atmos_var"), ) -model.load_checkpoint("microsoft/aurora", "aurora-0.25-pretrained.ckpt") +model.load_checkpoint("microsoft/aurora", "aurora-0.25-pretrained.ckpt", strict=False) # Normalisation means: locations["new_surf_var"] = 0.0