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文件里修改