关于导入权重的问题

siriusdemon / Build-Your-Own-Face-Model

Build your own ArcFace, CenterFace/Centernet

334 stars 69 forks source link

关于导入权重的问题 #25

Closed Heathcliff-saku closed 2 years ago

Heathcliff-saku commented 2 years ago

hi！我尝试 test.py 代码加载您的24.pth权重但是可能训练环境不一样（pytorch版本不同）出现报错：

RuntimeError: Error(s) in loading state_dict for DataParallel: Missing key(s) in state_dict: "module.conv1.net.0.net.0.bias", "module.conv2.net.0.bias", ... ,... (后面一堆)

请问up有好的解决方法或者在torch1.6版本以上训练的权重文件分享吗？

Heathcliff-saku commented 2 years ago

解决了，原来是我在模型里把bias开了。。。