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

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

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

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

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