CUDA error: device-side assert triggeredCUDA kernel errors might be asynchronously reported...

最近在使用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.

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值