This commit is contained in:
Glenn Jocher 2018-11-05 23:28:10 +01:00
parent 3afb29ad48
commit 19ccb41eaf
1 changed files with 1 additions and 1 deletions

View File

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