We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9c78de8 commit d7bab8aCopy full SHA for d7bab8a
timm/models/helpers.py
@@ -44,7 +44,7 @@ def load_state_dict(checkpoint_path, use_ema=False):
44
raise FileNotFoundError()
45
46
47
-def load_checkpoint(model, checkpoint_path, use_ema=False, strict=False):
+def load_checkpoint(model, checkpoint_path, use_ema=False, strict=True):
48
state_dict = load_state_dict(checkpoint_path, use_ema)
49
model.load_state_dict(state_dict, strict=strict)
50
0 commit comments