解决RuntimeError: CUDA error: device-side assert triggeredCUDA kernel errors…CUDA_LAUNCH_BLOCKING=1
在做对比实验运行其他算法的时候,发现报这个错,找了一下,是实际classes和模型最后一层的输出不一样
【无标题】
最新推荐文章于 2024-08-12 14:30:00 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)
解决RuntimeError: CUDA error: device-side assert triggeredCUDA kernel errors…CUDA_LAUNCH_BLOCKING=1
在做对比实验运行其他算法的时候,发现报这个错,找了一下,是实际classes和模型最后一层的输出不一样