YOLOv4-pytorch
YOLOv4-pytorch copied to clipboard
在模型的验证中出现了以下问题,应怎样解决呢? 谢谢!ValueError: only one element tensors can be converted to Python scalars
Val batch: 0%| | 0/12 [00:06<?, ?imgs/s]
Traceback (most recent call last):
File "/home/桌面/YOLOv4-pytorch/train.py", line 260, in
@jingtianyilong @
you should validation parameter: batch size is 1. change validation batch size parameter