2022年2月16日
记一次CUDA报错
摘要： 报错内容：CUDA error: device-side assert triggered 原因：使用ResNet50训练时使用了pretrained=True的模型，但实际类别数classes远超过MNIST自带的1000上限，导致标签越界。 解决方法：改成使用pretrained=False的模 阅读全文
posted @ 2022-02-16 13:56 Oberon 阅读(83) 评论(0) 推荐(0) 编辑