admin管理员组文章数量:1026989
RuntimeError
在跑模型的时候,出现了报错:
RuntimeError: Found dtype Long but expected Float
报错是发生在BCE loss 的部分,因为crossentropyloss要求batch_label必须为torch.float类型
所以需要修改一下标签,转为float格式就好了:
参考:
【RuntimeError: Found dtype Long but expected Float】
RuntimeError
在跑模型的时候,出现了报错:
RuntimeError: Found dtype Long but expected Float
报错是发生在BCE loss 的部分,因为crossentropyloss要求batch_label必须为torch.float类型
所以需要修改一下标签,转为float格式就好了:
参考:
【RuntimeError: Found dtype Long but expected Float】
本文标签: RuntimeError
版权声明:本文标题:RuntimeError 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:http://it.en369.cn/jiaocheng/1700134334a348734.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论