admin 管理员组

文章数量: 887021

YOLOv5训练报错：indices.append((b, a, gj.clamp

报错：

build target indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1))) # image, anchor, grid indices runtimeerror: result type float can’t be cast to the desired output type long int

源代码：

gain = torch.ones(7, device = device)

在loss.py文件里

解决方案：

gain = torch.ones(7, device = device).long()

在loss.py文件里修改

本文标签： YOLOv5训练报错indicesappend((b A gjclamp

版权声明：本文标题：YOLOv5训练报错：indices.append((b, a, gj.clamp 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.freenas.com.cn/jishu/1730916054h1406034.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。