Pytorch神经网络保存和加载注意事项
初学pytorch,记录一下自己踩过的坑。
最初的学习教程来自sofasofa:http://sofasofa.io/tutorials/pytorch_tutorials/
错误提示:
RuntimeError: Error(s) in loading state_dict for Sequential:
Missing key(s) in state_dict: "0.weight", "0.bias", "2.weight", "2.bias".
教程完整的代码
import to...
原创
2020-11-22 22:31:36 ·
707 阅读 ·
0 评论