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

在训练网络报错,错误如下:
在这里插入图片描述
报错原因:因为换了数据集,原本是CIFAR10数据集,CIFAR10数据集是10类,后来换了CIFAR100数据集,CIFAR100数据集为100类,因此换后要将对应的网络结构中的num_classes改成100,匹配后才会成功运行。

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

你可能感兴趣的:(深度学习,pytorch,神经网络,python)