最近在使用RESNET50的预训练模型迁移学习(transfer learning),因为RESNET50的最后的连接层out_features输出是1000,我这里疏忽写的out_features=5
,在开始训练的时候,我用的是CIFAR10数据集,是一个十分类的,只要label超过了5就会报这个错,因为图片对应的target超过5,在经过全连接层的时候,输出大于out_features。
是个小白,刚刚接触,记录一下!
RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.