minh9972t12 commited on
Commit
a17801d
·
1 Parent(s): a33f3d5

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +1 -1
config.yaml CHANGED
@@ -20,7 +20,7 @@ detection:
20
  13: broken_lamp
21
  model:
22
  confidence: 0.2
23
- device: cuda:0
24
  iou_threshold: 0.35
25
  path: models_large/best.pt
26
  visualization:
 
20
  13: broken_lamp
21
  model:
22
  confidence: 0.2
23
+ device: cpu
24
  iou_threshold: 0.35
25
  path: models_large/best.pt
26
  visualization: