admin管理员组文章数量:1794759
RuntimeError: CUDA error: device
在训练网络报错,错误如下: 报错原因:因为换了数据集,原本是CIFAR10数据集,CIFAR10数据集是10类,后来换了CIFAR100数据集,CIFAR100数据集为100类,因此换后要将对应的网络结构中的num_classes改成100,匹配后才会成功运行。
你的数据集多少类,你就得改成多少类
本文标签: CUDARuntimeErrordeviceerror
版权声明:本文标题:RuntimeError: CUDA error: device 内容由林淑君副主任自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:http://www.xiehuijuan.com/baike/1686554871a81314.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论