使用BCELoss出现RuntimeError: Found dtype Long but expected Float

最新推荐文章于 2025-02-18 15:30:11 发布

there2belief

最新推荐文章于 2025-02-18 15:30:11 发布

阅读量1k

点赞数 1

分类专栏： AI/ML/DL 文章标签：人工智能

原文链接：https://blog.csdn.net/znevegiveup1/article/details/124778676

版权

AI/ML/DL 专栏收录该内容

254 篇文章

订阅专栏

本文介绍了一种在使用PyTorch的BCELoss损失函数时遇到的数据类型不匹配问题及解决方案。当尝试将交叉熵损失函数替换为BCELoss时，由于预期目标张量类型为float而非int，导致运行时错误。文章提供了调整目标张量类型的代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

今天在跑模型的时候，发现了以下的报错

RuntimeError: Found dtype Long but expected Float

代码部分只是将crossentropyloss换成了BCELoss，那么这里的报错就发生在BCELoss的部分

loss_fn = torch.nn.BCELoss()
logit = logit.view(logit.size()[0]*logit.size()[1],-1)
batch_label = batch_label.view(batch_label.size()[0]*batch_label.size()[1],-1)
crossentropyloss = loss_fn(logit,batch_label)

crossentropyloss要求batch_label必须为torch.int类型，这里BCELoss的batch_label类型为torch.float
因此这段代码需要修改一下：

loss_fn = torch.nn.BCELoss()
logit = logit.view(logit.size()[0]*logit.size()[1],-1)
batch_label = batch_label.view(batch_label.size()[0]*batch_label.size()[1],-1)
batch_label = batch_label.to(torch.float)
crossentropyloss = loss_fn(logit,batch_label)

————————————————
来自：https://blog.csdn.net/znevegiveup1/article/details/124778676