diff --git a/train.py b/train.py index ce086f7c..15f412a9 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 = True + # torch.backends.cudnn.benchmark = True def main(opt):