首页 > 编程知识文章详情

RuntimeError: CUDA error: device-side ...For debugging consider passing CUDA_LAUNCH_BLOCKING=1

编程知识更新时间:2023-05-02 05:27:02

在训练网络报错,错误如下：

报错原因：因为换了数据集，原本是CIFAR10数据集，CIFAR10数据集是10类，后来换了CIFAR100数据集，CIFAR100数据集为100类，因此换后要将对应的网络结构中的num_classes改成100，匹配后才会成功运行。

你的数据集多少类，你就得改成多少类

更多推荐

RuntimeError: CUDA error: device-side ...For debugging consider passing CUDA_LAU

本文发布于:2023-04-26 06:12:00，感谢您对本站的认可！

本文链接:https://www.elefans.com/category/jswz/7e4590fb8ad5e3b98089c2d31b7cf74b.html

版权声明:本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系，我们将在24小时内删除。

error device RuntimeError CUDA passing

上一篇： Hikaril过一段时间就出错，Possibly consider using a shorter maxLifetime value问题
下一篇：返回列表

发布评论取消回复

评论列表（有 0 条评论）

热门文章