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):