updates

2019-08-06 17:44:09 +02:00 · 2019-08-06 17:44:09 +02:00 · 33601ca758
parent 01bc76faeb
commit 33601ca758
1 changed files with 1 additions and 1 deletions
--- 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