updates

2019-06-24 14:19:20 +02:00 · 2019-06-24 14:19:20 +02:00 · a3fcf20385
commit a3fcf20385
parent 1827b79647
1 changed files with 1 additions and 0 deletions
--- 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