Tencent / ObjectDetection-OneStageDet

单阶段通用目标检测器

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

问题报告

Yumingyuan opened this issue · comments

您好,我在运行train.py时出现一个奇怪的问题
File " ./vedanet/loss/_ regionloss.py", line 178,in forward
self . info['class'] = class prob.sum() . item() / obj cur
RuntimeError: CUDA error: device-side assert triggered
(python367) root@**********:~/code/0bjectDetection OneStageDet/yolo#

我也出现了同样的问题,请问有解决吗?