pytorch criterion踩坑小结

1. 数据类型不匹配：

报错：Expected object of type torch.LongTensor but found type torch.FloatTensor for argument #2 ‘target’

criterion = nn.CrossEntropyLoss()
loss = criterion(y_pre, y_train)

这里的y_train类型一定要是LongTensor的，所以在写DataSet的时候返回的label就要是LongTensor类型的，如下

def__init__(self, ...):self.label = torch.LongTensor(label)

2.target要用类标

报错：multi-target not supported at c:\new-builder_2\win-wheel\pytorch\aten\src\thnn\generic/ClassNLLCriterion.c:21

criterion = nn.CrossEntropyLoss()
loss = criterion(y_pre, y_train)

这里的y_train不能用one-hot编码，要用类标值。

pytorch criterion踩坑小结相关推荐

anaconda+cuda+cudnn+pytorch安装踩坑大全
windows环境下anaconda+pycharm+cuda+cudnn+pytorch安装踩坑大全 anaconda安装 pycharm安装 cuda cudnn pytorch gym环境安装 ...
树莓派下安装pytorch（踩坑记录）
目标:在树莓派4b上安装conda运行pytorch 硬件平台:树莓派4b 操作系统:Raspberry Pi OS with desktop 环境:pytorch 1. 安装系统镜像下载树莓派镜像 ...
ASP.NET MVC升级到ASP.NET Core MVC踩坑小结
写在前面 ASP.NET Core是微软新推出的支持跨平台.高性能.开源的开发框架,它的优势不必多说,因为已经说得太多了.当然,现在依然有着数量庞大的系统运行于.NET Framework上,由于有大 ...
戴尔游匣g15 5515笔记本电脑g3060锐龙cpu版装入显卡驱动配置pytorch环境踩坑总结
游戏本ubuntu21.04装机配置pytorch环境大全 ubuntu更换镜像源 0.删除之前的内容 1.开始的ubuntu20.04尝试 2.转换思路,改变系统的版本,使用ubuntu18.04 ...
Nvidia显卡主机安装Ubuntu系统踩坑小结（Z1-Entry-Tower-G6主机安装Ubuntu-18.04）
目录背景问题和解决方法 1.U盘安装Ubuntu系统时出现花屏和ACPI error问题: 2.安装Ubuntu系统时卡在Ubuntu logo五个圆点加载处: 3.系统安装成功后重启,出现花屏, ...
pytorch .item_从数据到模型，你可能需要1篇详实的pytorch踩坑指南
原创 · 作者 | Giant 学校 | 浙江大学研究方向 | 对话系统.text2sql 熟悉DL的朋友应该知道Tensorflow.Pytorch.Caffe这些成熟的框架,它们让广大AI爱好者 ...
Pytorch可视化工具tensorboardX（安装不踩坑）
安装方法(如何不踩坑,已修改): 首先默认你已经安装好pytorch(如果木有,为什么点进来呢) 安装tensorboardX pip install tensorboardX 安装tensorflo ...
pytorch转onnx踩坑日记
在深度学习模型部署时,从pytorch转换onnx的过程中,踩了一些坑.本文总结了这些踩坑记录,希望可以帮助其他人. 首先,简单说明一下pytorch转onnx的意义.在pytorch训练出一个深度学 ...
Pytorch踩坑记录：关于用net.eval()和with no grad装饰器计算结果不一样的问题
Pytorch踩坑记录相同点 net.eval()和with toch.no_grad()的相同点:都停止反向传播不同点: 1.net.eval() 用net.eval(),此时BN层会用训练时的 ...

pytorch criterion踩坑小结

1. 数据类型不匹配：

2.target要用类标

pytorch criterion踩坑小结相关推荐

最新文章

热门文章