最新消息:雨落星辰是一个专注网站SEO优化、网站SEO诊断、搜索引擎研究、网络营销推广、网站策划运营及站长类的自媒体原创博客

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

运维笔记admin16浏览0评论

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

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

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

发布评论

评论列表(0)

  1. 暂无评论