博客讲述了在运行Pytorch模型时，BCE loss部分出现报错的情况。原因是crossentropyloss要求batch_label必须为torch.float类型，解决办法是将标签转为float格式，还给出了相关参考链接。
在跑模型的时候，出现了报错：
RuntimeError: Found dtype Long but expected Float
报错是发生在BCE loss 的部分，因为crossentropyloss要求batch_label必须为torch.float类型
所以需要修改一下标签，转为float格式就好了：
参考：
【RuntimeError: Found dtype Long but expected Float】
https://blog.csdn.net/znevegiveup1/article/details/124778676
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
