![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
机器学习
小珈猫
这个作者很懒,什么都没留下…
展开
-
【代码笔记】RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should b
初学者经常会遇到下图所示的问题:RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same这是说 我们输入的数据类型与网络参数的类型不符。Input type为torch.cuda.FloatTensor(GPU数据类型)weight type(即net.parameters)为torch.FloatTensor(CPU数据类型)解决方法有三种:原创 2020-05-13 22:16:26 · 2387 阅读 · 1 评论 -
【代码笔记】解决加载模型错误方法总结RuntimeError: Error(s) in loading state_dict for Net: Missing key(s)
加载模型错误:有以下几种方法:1.在载入模型参数前加上:model = nn.DataParallel(model) 2.更改torch版本部分原因是模型之间torch版本不匹配导致,加载使用模型时和训练模型时的环境不一致,可以重新安装torch3.增加load_state_dict()属性model.load_state_dict(checkpoint['state_dict'], strict=False)...原创 2020-05-12 13:42:38 · 6758 阅读 · 6 评论