Merge pull request #32 from microsoft/wesselb/minor-clarifications

Minor clarifications
microsoft · Sep 13, 2024 · 8fecaee · 8fecaee
2 parents 171613d + 109600f
commit 8fecaee
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git 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
@@ -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