报错汇总
报错汇总,针对深度学习pytorch
AI 黎明
这个作者很懒,什么都没留下…
展开
-
TypeError: batch must contain tensors, numbers, dicts or lists;
参考:https://blog.csdn.net/zkp_987/article/details/81078923 核心错误出现在没有把图片transform为tensor from torchvision import transforms as T T.ToTensor()原创 2020-01-14 11:41:50 · 1470 阅读 · 1 评论 -
IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)
参考:http://www.seotest.cn/jishu/57263.html 问题背景:两个不一样的神经网络,读取数据和训练模型等代码一样,只是两个模型不一样。经过分析问题一定出在模型上 原代码: def forward(self, input): x = self.feature(input) x = x.squeeze() fc = ...原创 2020-01-07 16:15:49 · 21653 阅读 · 2 评论