YOLOv6 RuntimeError: all elements of input should be between 0 and 1

RuntimeError: all elements of input should be between 0 and 1

Open joel5638 opened this issue 2 years ago • 2 comments

Im trying with 450 images in train and 130 in val. I have the training started but encountered this error.

RuntimeError: all elements of input should be between 0 and 1

Im attaching a screenshot of the error too Screenshot from 2022-07-01 22-11-36

Jul 01 '22 16:07 joel5638

Hi, did you set the CUDA environment correctly. If your computer does not have GPU, you can try to add --device=cpu when you run the training code.

Jul 03 '22 05:07 mtjhl

@mtjhl Im training using cpu as my device

Jul 12 '22 12:07 joel5638