updates
This commit is contained in:
parent
6f777e2bc5
commit
4459a9474e
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue