为什么我的CUDA程序会突然报错,原因究竟是什么?
摘要:报错内容:CUDA error: device-side assert triggered 原因:使用ResNet50训练时使用了pretrained=True的模型,但实际类别数classes远超过MNIST自带的1000上限,导致标签越
报错内容:CUDA error: device-side assert triggered
原因:使用ResNet50训练时使用了pretrained=True的模型,但实际类别数classes远超过MNIST自带的1000上限,导致标签越界。
解决方法:改成使用pretrained=False的模型,并指定num_classes
