From 2463030d6c8d36d8c45770706825b57adf72ae93 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 8 Nov 2018 12:28:19 +0100 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 9d2b1a48..fbadd933 100644 --- a/train.py +++ b/train.py @@ -24,7 +24,7 @@ torch.manual_seed(0) if cuda: torch.cuda.manual_seed(0) torch.cuda.manual_seed_all(0) - torch.backends.cudnn.benchmark = False + torch.backends.cudnn.benchmark = True def main(opt):