Skip to content

Commit

Permalink
Add strict=False when adding new variables
Browse files Browse the repository at this point in the history
  • Loading branch information
wesselb committed Sep 13, 2024
1 parent ba42e43 commit 109600f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/finetuning.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 109600f

Please sign in to comment.