RuntimeError: view size is not compatible with input tensor's size and stride

[mikel-brostrom]

Got this error:

File "train.py", line 211, in build_target
    pred_ious = bboxes_iou(pred[b].view(-1, 4), truth_box, xyxy=False)
RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead.

Solved it by changing view:

pytorch-YOLOv4/train.py

Line 211 in a65d219

pred_ious = bboxes_iou(pred[b].view(-1, 4), truth_box, xyxy=False)

to reshape

[J1ia]

在view前面加.contiguous()