From fd6619d77371d76425e644ee02932be0bd71069b Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 3 Dec 2018 22:33:25 +0100
Subject: [PATCH] updates

---
 train.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/train.py b/train.py
index beb98933..93ac9c96 100644
--- 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):