updates

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