From a3fcf203856c1bc23666348d566826a420c95783 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 24 Jun 2019 14:19:20 +0200 Subject: [PATCH] updates --- detect.py | 1 + 1 file changed, 1 insertion(+) diff --git a/detect.py b/detect.py index 5defda55..192ef310 100644 --- a/detect.py +++ b/detect.py @@ -22,6 +22,7 @@ def detect( webcam=False ): device = torch_utils.select_device() + torch.backends.cudnn.benchmark = False # set False for reproducible results if os.path.exists(output): shutil.rmtree(output) # delete output folder os.makedirs(output) # make new output folder