返回顶部

深度学习模型训练时报错“nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Float‘问题解决

报错如下：

RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Float'

一般来说这个问题是计算Loss时的报错。

解决方法：

将如下代码

loss_func(torch.squeeze(output), target)

改为

loss_func(torch.squeeze(output), target.long())

 

posted @   欣杰科技  阅读(1988)  评论(0)  编辑  收藏  举报
相关博文：
·  四两拨千斤之R-Drop核心代码
·  深度学习模型训练中，输入数据inputs维度和标签数据lablels维度调整方法
·  RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘
·  Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
·  RuntimeError: CUDA error: device-side assert triggered
阅读排行：
· TypeScript + Deepseek 打造卜卦网站：技术与玄学的结合
· 阿里巴巴 QwQ-32B真的超越了 DeepSeek R-1吗？
· 【译】Visual Studio 中新的强大生产力特性
· 10年+ .NET Coder 心语 ── 封装的思维：从隐藏、稳定开始理解其本质意义
· 【设计模式】告别冗长if-else语句：使用策略模式优化代码结构
点击右上角即可分享
微信分享提示
AI IDE Trae