![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
文章平均质量分 77
溟玖
这个作者很懒,什么都没留下…
展开
-
pytorch分布式遇到的问题集锦
而使用这个语句来查看模型的权重时会发现返回的权重字典确实和需要加载的权重字典一样,这时我以为权重字典成功加载到模型net中了,但是使用加载了权重字典的模型net预测数据集时,却出现了模型预测性能和随机预测差不多,这说明模型其实并没有成功加载之前训练好的模型的权重字典。,发现返回的权重字典和模型初始权重字典一样,这解释了为什么模型预测性能和随机预测差不多了,因为模型确实并未成功加载权重字典,起码分布式的模型并未成功加载权重字典。pytorch中模型可以调用一个state_dict()函数查看权重字典的。原创 2023-12-13 19:29:20 · 1107 阅读 · 1 评论 -
学习pytorch时常见错误集锦
通常是因为数据集的分类标签(label)中存在不合理的标签。比如:n分类的标签应该是range(0, n),但是数据集中却出现了其他标签。:可能是因为 y 不是整型,因为在模型中传递的数据要求是Float型,但是。作为标签传递给损失函数的数据必须是Int型。如果排除了标签问题,请看。原创 2023-12-05 22:02:31 · 347 阅读 · 0 评论