updates

2018-12-03 22:33:25 +01:00
parent 10cca39934
commit fd6619d773
1 changed files with 2 additions and 1 deletions
@@ -35,6 +35,7 @@ torch.manual_seed(0)
 if cuda:
    torch.cuda.manual_seed(0)
    torch.cuda.manual_seed_all(0)
+    if not opt.multi_scale:
        torch.backends.cudnn.benchmark = True