From d26df074a6b96e921adf9f6859cc4ebb0e4f99ee Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 10 Sep 2019 14:24:46 +0200 Subject: [PATCH] updates --- utils/torch_utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/utils/torch_utils.py b/utils/torch_utils.py index 02543118..9bf97548 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -21,7 +21,6 @@ def select_device(force_cpu=False, apex=False): if not cuda: print('Using CPU') if cuda: - torch.backends.cudnn.benchmark = True # set False for reproducible results c = 1024 ** 2 # bytes to MB ng = torch.cuda.device_count() x = [torch.cuda.get_device_properties(i) for i in range(ng)]