result type Float can't be cast to the desired output type __int64
File "D:\git\YOLOv5-Lite-master\utils\loss.py", line 211, in build_targets
#indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1))) # image, anchor, grid indices
indices.append((b, a, gj.clamp_(0, int(gain[3]) - 1), gi.clamp_(0, int(gain[2]) - 1)))