yolov1: "Process finished with exit code 139 (interrupted by signal 11: SIGSEGV)"

[sunnysky29]

when I want to check the pic with the following code:
it gives me the wrong info.
and device is "cpu"

`
for epoch in range(EPOCHS):
for x, y in train_loader:
x = x.to(DEVICE)
for idx in range(8):
bboxes = cellboxes_to_boxes(model(x))
bboxes = non_max_suppression(bboxes[idx], iou_threshold=0.5, threshold=0.4, box_format="midpoint")
plot_image(x[idx].permute(1,2,0).to("cpu"), bboxes)

       import sys
       sys.exit()

`

so, the problem is realted to the "cpu"??