Skip to content

Commit 01b9108

Browse files
committed
Merge branch 'master' into more_attn
2 parents 02f9d4b + d7bab8a commit 01b9108

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def load_state_dict(checkpoint_path, use_ema=False):
4444
raise FileNotFoundError()
4545

4646

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

0 commit comments

Comments
 (0)