update cudnn benchmark setting

2026-09-20 01:31:13 +02:00 · 2023-05-14 12:28:37 -04:00
parent 760f5fb89a
commit 618a1703ae
1 changed files with 2 additions and 1 deletions
@@ -106,7 +106,8 @@ def set_cuda_params():
        if torch.backends.cudnn.is_available():
            try:
                torch.backends.cudnn.benchmark = True
-                torch.backends.cudnn.benchmark_limit = 0 if shared.opts.cudnn_benchmark else 1
+                if shared.opts.cudnn_benchmark:
+                    torch.backends.cudnn.benchmark_limit = 0
                torch.backends.cudnn.allow_tf32 = shared.opts.cuda_allow_tf32
            except:
                pass