This commit is contained in:
Glenn Jocher
2019-09-17 15:16:29 +02:00
parent 425ed4b84b
commit 78bb153d7d
2 changed files with 2 additions and 3 deletions
-1
View File
@@ -4,7 +4,6 @@ import torch
def init_seeds(seed=0):
torch.cuda.empty_cache()
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed)