Fix `KeyError` when initialize the model with `ignore_mismatched_sizes=True` #16920
use original loaded keys to find mismatched keys
e85f1f5f
tricktreat
changed the title use original loaded keys to find mismatched keys Use original loaded keys to find mismatched keys 4 years ago
tricktreat
changed the title Use original loaded keys to find mismatched keys Fix KeyError when initialize a model from pre-trained weights with ignore_mismatched_sizes=True. 4 years ago
tricktreat
changed the title Fix KeyError when initialize a model from pre-trained weights with ignore_mismatched_sizes=True. Fix KeyError when initialize the model from pre-trained weights with ignore_mismatched_sizes=True. 4 years ago
tricktreat
changed the title Fix KeyError when initialize the model from pre-trained weights with ignore_mismatched_sizes=True. Fix `KeyError` when initialize the model with `ignore_mismatched_sizes=True` 4 years ago
sgugger
approved these changes
on 2022-04-26
sgugger
merged
2d91e3c3
into main 4 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub