diff --git a/utils/torch_utils.py b/utils/torch_utils.py index d984dfca..d615e2a8 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -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: