dog-qiuqiu/Yolo-FastestV2

bug

Opened this issue · 0 comments

Traceback (most recent call last):
File "train.py", line 107, in
iou_loss, obj_loss, cls_loss, total_loss = utils.loss.compute_loss(preds, targets, cfg, device)
File "E:\Backup\Desktop\Yolo-Fast\utils\loss.py", line 147, in compute_loss
tcls, tbox, indices, anchors = build_target(preds, targets, cfg, device)
File "E:\Backup\Desktop\Yolo-Fast\utils\loss.py", line 120, in 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 __int64

pytorch 1.12.1