kadirnar commited on
Commit
e0b26b3
1 Parent(s): b65e7f9

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: cpu
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: cuda:0
10
  # F16 precision
11
  HALF: False
12
  # The path of the yolov6 label file