RuntimeError: cannot perform reduction function max on tensor with no elements because the operation does not have an identity

[rohanpawar294]

Search before asking

• I have searched the YOLOv5 issues and found no similar bug report.

YOLOv5 Component

Validation

Bug

File "train.py", line 630, in
main(opt)
File "train.py", line 527, in main
train(opt.hyp, opt, device, callbacks)
File "train.py", line 368, in train
compute_loss=compute_loss)
File "/home/griffserver2/yolo_env/lib/python3.6/site-packages/torch/autograd/grad_mode.py", line 26, in decorate_context
return func(*args, **kwargs)
File "/home/griffserver2/Rohan/yolov5_rotation-master/val.py", line 198, in run
out = non_max_suppression(out, conf_thres, iou_thres, labels=lb, multi_label=True, agnostic=single_cls)
File "/home/griffserver2/Rohan/yolov5_rotation-master/utils/general.py", line 697, in non_max_suppression
conf_angle, j_angle = x[i, 5+nc:].max(1, keepdim=True)
RuntimeError: cannot perform reduction function max on tensor with no elements because the operation does not have an identity

Environment

No response

Minimal Reproducible Example

No response

Additional

No response

Are you willing to submit a PR?

• Yes I'd like to help by submitting a PR!