RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THCUNN/gen

pytorch报错：RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THCUNN/generic/ClassNLLCriterion.cu:224原因：模型输出的类别个数和读入的类别个数不一致导致的将输出类别个数修改为与输入类别个数一致即可。...

zhuiyuan2012

517人浏览 · 2021-10-18 11:23:45

zhuiyuan2012 · 2021-10-18 11:23:45 发布

pytorch报错：

RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THCUNN/generic/ClassNLLCriterion.cu:224

原因：模型输出的类别个数和读入的类别个数不一致导致的

将输出类别个数修改为与输入类别个数一致即可。

https://edu.csdn.net/learn/39067/627173?utm_source=2019755004

汇聚全球AI编程工具，助力开发者即刻编程。

更多推荐

AI辅助编程工具是什么

最终转化为可直接运行的代码，并自动创建对应的文件结构、配置（JSON / YAML）、脚本（Shell）、以及自动化测试（Automated testing）。AWS Kiro 四档版本对比：从 Free 到 Power，分别覆盖日常开发、中型项目、多服务工程与团队级自动化，全链路提升代码生成、架构设计与自动部署能力。未来 3–5 年，AI辅助编程将成为所有开发者默认使用的基础设施，就像版本控制、