Fix strict flag change for checkpoint load.

4 years ago · d7bab8a6c5
parent 9c78de8c02
commit d7bab8a6c5
1 changed files with 1 additions and 1 deletions
--- a/timm/models/helpers.py
+++ b/timm/models/helpers.py
@ -44,7 +44,7 @@ def load_state_dict(checkpoint_path, use_ema=False):
        raise FileNotFoundError()


-def load_checkpoint(model, checkpoint_path, use_ema=False, strict=False):
+def load_checkpoint(model, checkpoint_path, use_ema=False, strict=True):
    state_dict = load_state_dict(checkpoint_path, use_ema)
    model.load_state_dict(state_dict, strict=strict)