admin管理员组

文章数量:1565829

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

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

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

本文标签: ErrordeviceRuntimeErrorCUDApassing