missing strict=False in util.misc.load_model - Githubissues

facebookresearch / mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

Other

7.2k stars 1.2k forks source link

missing strict=False in util.misc.load_model #121

Open eminorhan opened 2 years ago

eminorhan commented 2 years ago

I believe the load_state_dict function in this line should set strict=False: https://github.com/facebookresearch/mae/blob/efb2a8062c206524e35e47d04501ed4f544c0ae8/util/misc.py#L322

Otherwise, I get a missing keys error in this line (possibly elsewhere too): https://github.com/facebookresearch/mae/blob/efb2a8062c206524e35e47d04501ed4f544c0ae8/main_linprobe.py#L260