From b5a9007c70d5c4e91a9da49ac11a862e48898fca Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Wed, 31 Jan 2024 15:31:34 +0100 Subject: [PATCH] Try being explicit --- torchgeo/trainers/utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/torchgeo/trainers/utils.py b/torchgeo/trainers/utils.py index 7a54f936cb4..b5cd8f1e923 100644 --- a/torchgeo/trainers/utils.py +++ b/torchgeo/trainers/utils.py @@ -117,7 +117,10 @@ def load_state_dict( state_dict[output_module_key + ".bias"], ) - return model.load_state_dict(state_dict, strict=False) + missing_keys: list[str] + unexpected_keys: list[str] + missing_keys, unexpected_keys = model.load_state_dict(state_dict, strict=False) + return missing_keys, unexpected_keys def reinit_initial_conv_layer(