Hugging Face's logo

Spaces:
kadirnar
/
torchyolo
Runtime error

App Files Community
1
kadirnar commited on
Commit
b411301
·
1 Parent(s): ef091ee

Update default_config.yaml

Browse files
Files changed (1) hide show
  1. default_config.yaml +1 -1
default_config.yaml CHANGED
@@ -6,7 +6,7 @@ DETECTOR_CONFIG:
6
  # The size of the image
7
  IMAGE_SIZE: 640
8
  # The device to run the detector
9
- DEVICE: cuda:0
10
  # F16 precision
11
  HALF: False
12
  # The path of the yolov6 label file
 
6
  # The size of the image
7
  IMAGE_SIZE: 640
8
  # The device to run the detector
9
+ DEVICE: cpu
10
  # F16 precision
11
  HALF: False
12
  # The path of the yolov6 label file