updates

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


 def main(opt):