Update nnfabrik/utility/nn_helpers.py

Co-authored-by: Max Burg <[email protected]>
sinzlab · Apr 8, 2024 · d07ff04 · d07ff04
1 parent 9c5ca97
commit d07ff04
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nnfabrik/utility/nn_helpers.py b/nnfabrik/utility/nn_helpers.py
@@ -158,7 +158,7 @@ def load_state_dict(
             stripped_state_dict[new_key] = v
         state_dict = stripped_state_dict
 
-    if strict is None:
+    if strict:
         # 1. filter out missing keys
         filtered_state_dict = {k: v for k, v in state_dict.items() if k in model_dict}
         unused = set(state_dict.keys()) - set(filtered_state_dict.keys())