Fix strict flag change for checkpoint load.

d7bab8a6 · Ross Wightman · 9c78de8c · d7bab8a6
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

timm/models/helpers.py timm/models/helpers.py +1 -1

未找到文件。
--- 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)