记一次CUDA报错

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

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

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

posted @ 2022-02-16 13:56  Oberon  阅读(87)  评论(0)  编辑  收藏  举报