YOLOv4-pytorch icon indicating copy to clipboard operation
YOLOv4-pytorch copied to clipboard

在模型的验证中出现了以下问题,应怎样解决呢? 谢谢!ValueError: only one element tensors can be converted to Python scalars

Open lifafang opened this issue 3 years ago • 2 comments

Val batch: 0%| | 0/12 [00:06<?, ?imgs/s] Traceback (most recent call last): File "/home/桌面/YOLOv4-pytorch/train.py", line 260, in Trainer(log_dir,resume= args.resume).train() File "/home/桌面/YOLOv4-pytorch/train.py", line 171, in train coco_stat = evaluator.evaluate(self.yolov4) File "/home/桌面/YOLOv4-pytorch/eval/cocoapi_evaluator.py", line 61, in evaluate info_img = [float(info) for info in info_img] File "/home/桌面/YOLOv4-pytorch/eval/cocoapi_evaluator.py", line 61, in info_img = [float(info) for info in info_img] ValueError: only one element tensors can be converted to Python scalars

lifafang avatar Jul 27 '21 12:07 lifafang

@jingtianyilong @

lifafang avatar Jul 27 '21 14:07 lifafang

you should validation parameter: batch size is 1. change validation batch size parameter

kimsjpk1 avatar Jan 10 '22 07:01 kimsjpk1