RuntimeError: CUDA error: device
在训练网络报错,错误如下： 报错原因：因为换了数据集，原本是CIFAR10数据集，CIFAR10数据集是10类，后来换了CIFAR100数据集，CIFAR100数据集为100类，因此换后要将对应的网络结构中的num_classes改成100，匹配后才会成功运行。
你的数据集多少类，你就得改成多少类
👁️ 阅读量：0
© 版权声明：本文《RuntimeError: CUDA error: device》内容均为本站精心整理或网友自愿分享，如需转载请注明原文出处：https://www.zastudy.cn/wen/1686555090a290694.html。