apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Glenn Jocher
2018-12-03 22:33:25 +01:00
parent 10cca39934
commit fd6619d773
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+2 -1
View File
@@ -35,7 +35,8 @@ torch.manual_seed(0)
if cuda: if cuda:
torch.cuda.manual_seed(0) torch.cuda.manual_seed(0)
torch.cuda.manual_seed_all(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): def main(opt):