admin管理员组

文章数量:1414693

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.

问题：一般是类别不一致导致的，网络输出类别跟输入数据类别不一致导致，我的数据集是15各类别，网络输出类别设置的14不一致导致的。

解决方法：确认数据集种类别数，修改网络输出类别数

我的代码修改部分如下所示，将num_classes修改为15：

model.head.postprocessor.iou_thr = 0.6
model.head.postprocessor.score_thr = 0.001
model.head.postprocessor.multi_label = True
model.head.headmodule.num_classes = 15

本文标签： 无标题DebuggingpassingCUDALAUNCHBLOCKING

版权声明：本文标题：【无标题】For debugging consider passing CUDA_LAUNCH_BLOCKING=1. 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.betaflare.com/biancheng/1738332661a2076330.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。