yolov5训练出现RuntimeError: result type Float can't be cast to the desired output type __int64

这个错误是由于数据类型转换错误引起的。当尝试将Float类型的结果转换为__int64类型时，会出现这个错误。为了解决这个问题，你可以尝试修改对应的代码，使用正确的数据类型进行转换。修改gain的类型为torch.ones(7, device=targets.device).float()，这样就可以将Float类型的结果正确地转换为__int64类型。