The model and loaded state dict do not match exactly

SamsungLabs / iterdet

[S+SSPR2020] IterDet: Iterative Scheme for Object Detection in Crowded Environments

https://arxiv.org/abs/2005.05708

Mozilla Public License 2.0

210 stars 39 forks source link

The model and loaded state dict do not match exactly #5

Closed xiaocc612 closed 4 years ago

xiaocc612 commented 4 years ago

Thanks for sharing your work! When I follow the *Get Started** instruction, I met the following error: 2020-06-08 16:47:55,759 - root - INFO - load model from: torchvision://resnet50 2020-06-08 16:47:55,913 - mmdet - WARNING - The model and loaded state dict do not match exactly unexpected key in source state_dict: fc.weight, fc.bias

Can you help me to solve this?

filaPro commented 4 years ago

Hi, @xiaocc612 It is not an error, it is warning which is normal in case of loading of pretrained ResNet50 without last fully-connected layers. You can check same issue in mmdetection repo, ex. https://github.com/open-mmlab/mmdetection/issues/2175 .

xiaocc612 commented 4 years ago

thanks.