This commit is contained in:
Glenn Jocher 2020-01-15 12:27:54 -08:00
parent 6f777e2bc5
commit 4459a9474e
1 changed files with 0 additions and 2 deletions

View File

@ -5,8 +5,6 @@ import torch
def init_seeds(seed=0):
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed)
# Remove randomness (may be slower on Tesla GPUs) # https://pytorch.org/docs/stable/notes/randomness.html
if seed == 0: