Frankluox / LightningFSL

LightningFSL: Pytorch-Lightning implementations of Few-Shot Learning models.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Problem when reproducing Meta-baseline

LIUZIJING-CHN opened this issue · comments

Excuse me,
After I finished the pretraining of meta-baseline, I got an error when loading the pretrained model in finetune stage.
RuntimeError: Error(s) in loading state_dict for ProtoNet: Missing key(s) in state_dict "classifier.scale_cls". Unexpected key(s) in state_dict: "classifier.weight", "classifier.bias".
It seems that you may have a wrong setting in the file set_config_meta_baseline_finetune.py

Yes. Thanks for reporting the wrong setting. Please set config_dict["load_backbone_only"] in line 23 to True and see if everything goes well.

Thx, it works.