From 33601ca7582fccf4a789e75a42c38f6fcfb19457 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 6 Aug 2019 17:44:09 +0200
Subject: [PATCH] updates

---
 detect.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/detect.py b/detect.py
index ceaa86cf..5d0f2b4e 100644
--- a/detect.py
+++ b/detect.py
@@ -19,7 +19,7 @@ def detect(cfg,
            save_txt=False,
            save_images=True):
     # Initialize
-    device = torch_utils.select_device()
+    device = torch_utils.select_device(force_cpu=ONNX_EXPORT)
     torch.backends.cudnn.benchmark = False  # set False for reproducible results
     if os.path.exists(output):
         shutil.rmtree(output)  # delete output folder