Failed training custom data set using CPU

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

GNU General Public License v3.0

13.33k stars 4.2k forks source link

Failed training custom data set using CPU #495

Open choiryaldisp opened 2 years ago

choiryaldisp commented 2 years ago

Hello, I had tried to train custom data set using CPU, but i got message like this Anyone, can help me to solve it?

i had tried add --device cpu when running the program

/site-packages/torch/amp/autocast_mode.py:198: UserWarning: User provided device_type of 'cuda', but CUDA is not available. Disabling warnings.warn('User provided device_type of \'cuda\', but CUDA is not available. Disabling') Killed

mle-usti commented 2 years ago

try setting cpu = True at line#66 in utils/torch_utils.py

dazzafact commented 2 years ago

parser.add_argument('--device', default='cpu', help=......