updates

2019-09-10 14:24:46 +02:00 · 2019-09-10 14:24:46 +02:00 · d26df074a6
parent a2016201f3
commit d26df074a6
1 changed files with 0 additions and 1 deletions
--- 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)]