Merge branch 'master' into more_attn

4 years ago · 01b9108619
parent 02f9d4bc34 d7bab8a6c5
commit 01b9108619
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)