RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘解决办法

最新推荐文章于 2024-03-19 12:50:00 发布
最新推荐文章于 2024-03-19 12:50:00 发布
阅读量1.3w 收藏 11
点赞数 8
分类专栏： 深度学习 文章标签： pytorch 深度学习 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qi_sama/article/details/122402880
版权

运行pytorch在计算cross_entropy的loss遇到报错

RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int'`

原因是pytorch自带的损失计算函数不支持原本的标签1,2,3,4…,n,需要转换成网络需要的one_hot编码才行
具体代码如下

 one_hot = torch.zeros(np.array(batch_size, num_class, device=torch.device('cuda:0')).scatter_(1, label, 1)

batch_size和num_class视自身数据集情况而定，label数据集格式为[batch_size,1]，如果不是二维，那么改为label.unsqueeze(1)，数据格式要是int64才能被scatter函数调用

qi_sama
关注
  • 8
    点赞
  • 踩
  • 11
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
专栏目录
RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘
WYKB_Mr_Q的博客
03-27 5655
Traceback (most recent call last): File "E:/relate_code/Gaitpart-master/main.py", line 149, in <module> train_model() File "E:/relate_code/Gaitpart-master/main.py", line 98, in train_model loss = criterion(outputs, labels) File "C:\Anac
CANopen_for_3S_Runtime_Systems_V2_3_5_0.zip_CANOPEN PLC_CODESYS
07-15
用于PLC编程，用于codesys编程，帮助初学者尽量掌握编程技巧
Pytorch报错`RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Float‘`
最新发布
qq_43369406的博客
03-19 280
出现这个原因是因为我的代码中使用了BCE Loss，而BCE Loss函数输入的label标签类型应该是。
aws-lambda-haskell-runtime：:high_voltage:AWSLambda的Haskell运行时
01-30
aws-lambda-haskell-runtime：:high_voltage:AWSLambda的Haskell运行时
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
01-20
跑模型时出现RuntimeError: CUDA out of memory.错误 查阅了许多相关内容，原因是：GPU显存内存不够 简单总结一下解决方法： 将batch_size改小。 取torch变量标量值时使用item()属性。 可以在测试阶段添加如下代码： with torch.no_grad(): # 停止自动反向计算梯度 参考： https://wisdomai.xyz/tool/pytorch/archives/2301 https://ptorch.com/news/160.html 作者：菜叶儿掉啦
SIMATIC_WinCC_Runtime_Advanced_V16_安装包_链接地址baiduyun.txt
01-28
SIMATIC_WinCC_Runtime_Advanced_V16_安装包_链接地址
[debug] RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘
真理无穷，进一步有进一步的欢喜
01-11 1万+
问题描述 在使用nn.CrossEntropyLoss时出现报错： loss_id = self.IDLoss(id_preds, id_targets) Traceback (most recent call last): File "/home/hgx/pycharm-2020.1.3/plugins/python/helpers/pydev/_pydevd_bundle/pydevd_exec2.py", line 3, in Exec exec(exp, global_vars, lo
【已解决】RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Float‘
BetrayFree的博客
10-20 4953
来进行训练模型加水印时报错RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Float'，这个在网上并没有找到解决办法，有的也只是RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'int'，也就是说之前因为int类型不兼容出了问题，那这个float报错的原因是什么呢？
pytorch自己收集的数据模型训练遇到的几个问题
sslc_的博客
08-19 1929
pytorch模型训练遇到的几个问题
emf-sdo-runtime-2.1.1.zip_EMF Runtime Plugin_Eclipse_eclipse ros
09-23
eclipse rose 插件，用于eclipse生成UML图。
解决pycharm导入numpy包的和使用时报错：RuntimeError: The current Numpy installation (‘D:\\python3.6\\lib\\site-packa的问题
01-21
标题用pycharm导入numpy包的和使用时报错：RuntimeError: The current Numpy installation (‘D:\python3.6\lib\site-packages\numpy\init.py 1.file→settings→project interpreter→+（建议用pychon3.6版本，我之前用3.8版本安装不上numpy）， 2.搜索numpy,注意把下面对号点上 3.现在简单用numpy还是会报错：RuntimeError: The current Numpy installation (‘D:\python3.6\lib\site
Python RuntimeError: thread.__init__() not called解决方法
09-22
主要介绍了Python RuntimeError: thread.__init__() not called解决方法,需要的朋友可以参考下
RuntimeError: _thnn_nll_loss_forward not supported on CUDAType for Long
风口IT猪的成长录
08-07 1601
RuntimeError: _thnn_nll_loss_forward not supported on CUDAType for Long报错内容解决方案 报错内容 解决方案 更改报错位置tensor的数据类型 loss = F.nll_loss(decoder_outputs.double(), target, ignore_index=config.num_sequence.PAD) # 带权损失 即可搞定！ 参考：link ...
解决RuntimeError: _thnn_mse_loss_forward is not implemented for type torch.cuda.LongTensor和scatter_方法
热门推荐
tanglinjie的CSDN博客
02-26 1万+
在PyTorch中遇到了如标题的问题，网上大多数给的是类型不匹配问题，在stackoverflow找到了问题的答案，这里出现的问题是因为loss需要one-hot类型的数据，而我们使用的是类别标签。 什么是one-hot？ 一个例子解释什么是one-hot，对于5分类问题，我们使用[0,0,1,0,0]来表示这个实例是属于第三个类别的，等价于类别标签[2]（从0对类别编码）。关于one-hot的好...
{RuntimeError}“nll_loss_forward_reduce_cuda_kernel_1d_index“ not implemented for ‘Float‘
weixin_39417324的博客
12-27 8513
应该是使用损失函数的时候，遇到了这个问题，意思就是说，这个函数的某个参数不支持Float类型的： F.nll_loss(out, target) 这个函数就是算损失，一般来说，这个函数使用应该遵循下面两点： 第一点，应该前后维度一致，如果你的batchsize大于1，那么可以都展开成为一维的 第二点，out的类型是cuda类型的float，target是long()类型的 F.nll_loss(out.squeeze().flatten(), target_class.flatten().long())
“nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘ 问题解决
weixin_45218778的博客
02-15 8572
错误nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int'，这是pytorch在计算交叉熵时出现的错误
RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int'
05-26
这个错误通常是由于数据类型不匹配导致的。在 PyTorch 中，`nll_loss` 函数的输入参数通常是浮点数类型的张量，而不是整数类型的张量。 你可以尝试将输入张量转换为浮点数类型，例如使用 `.float()` 方法将整数类型的张量转换为浮点数类型的张量： ```python import torch # 创建一个整数类型的张量 x = torch.tensor([1, 2, 3], dtype=torch.int) # 将整数类型的张量转换为浮点数类型的张量 x = x.float() # 使用 nll_loss 函数对张量进行计算 loss = torch.nn.functional.nll_loss(x, ...) ``` 如果你仍然遇到问题，请提供更多的上下文和代码，以便更好地帮助你解决问题。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值