记一次CUDA报错
报错内容:
CUDA error: device-side assert triggered
原因:使用ResNet50训练时使用了pretrained=True
的模型,但实际类别数classes
远超过MNIST自带的1000上限,导致标签越界。
解决方法:改成使用pretrained=False
的模型,并指定num_classes
报错内容:
CUDA error: device-side assert triggered
原因:使用ResNet50训练时使用了pretrained=True
的模型,但实际类别数classes
远超过MNIST自带的1000上限,导致标签越界。
解决方法:改成使用pretrained=False
的模型,并指定num_classes