training

[sunriselqq]

Hello! When I training your sample dataset, it suffered an error. The printed information is as follow:

Traceback (most recent call last):
File "rotate_train.py", line 557, in
train(hyp, opt, device, tb_writer, rotate=opt.rotate)
File "rotate_train.py", line 318, in train
loss, loss_items = compute_loss(pred, targets.to(device)) # loss scaled by batch_size
File "E:\RotateObjectDetection\rotate-yolov5\utils\loss.py", line 269, in call
iou = rotate_bbox_iou(pbox, tbox[i], CIoU=True, device=device) # iou(prediction, target)
File "E:\RotateObjectDetection\rotate-yolov5\utils\general.py", line 1070, in rotate_bbox_iou
boxes1_xyxyxyxy = xywhrm2xyxyxyxy(boxes1)
File "E:\RotateObjectDetection\rotate-yolov5\utils\general.py", line 985, in xywhrm2xyxyxyxy
xyxyxyxy = torch.matmul(xyxyxyxy, R).view(-1, 8).contiguous()+xywhrm[:, [0, 1, 0, 1, 0, 1, 0, 1]]
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)

I modified the device to cuda "0", it did not work.

[XinzeLee]

Hi, I think it has nothing to do with device id. You can specify CUDA_LAUNCH_BLOCKING=1 to view the underlying reason. If this cannot give you hint about how to solve it, you may need to follow the tutorials in README.md and Rotate-Tutorial1.ipynb again step by step to see where is the problem.