From cca17f4d1e1c1e1ce4529619bef7dad0c1001e8a Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 4 Sep 2019 14:49:10 +0200
Subject: [PATCH] updates

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
---
 detect.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/detect.py b/detect.py
index 9ee68453..b9215967 100644
--- a/detect.py
+++ b/detect.py
@@ -14,6 +14,7 @@ def detect(save_txt=False, save_img=False, stream_img=False):
 
     # Initialize
     device = torch_utils.select_device(force_cpu=ONNX_EXPORT)
+    torch.backends.cudnn.benchmark = False  # set True to speed up constant image size inference
     if os.path.exists(out):
         shutil.rmtree(out)  # delete output folder
     os.makedirs(out)  # make new output folder
@@ -29,7 +30,6 @@ def detect(save_txt=False, save_img=False, stream_img=False):
 
     # Fuse Conv2d + BatchNorm2d layers
     # model.fuse()
-    # torch.backends.cudnn.benchmark = True  # set True to speed up constant image size inference
 
     # Eval mode
     model.to(device).eval()