记一次CUDA报错


报错内容:CUDA error: device-side assert triggered

原因:使用ResNet50训练时使用了pretrained=True的模型,但实际类别数classes远超过MNIST自带的1000上限,导致标签越界。

解决方法:改成使用pretrained=False的模型,并指定num_classes