-
-
Notifications
You must be signed in to change notification settings - Fork 692
DeterministicEngine improvement #2754
Copy link
Copy link
Closed
Labels
Description
- remove
torch.cuda.manual_seed_all(seed)here astorch.manual_seed(seed)will seed CUDA if it has CUDA enabled. - try to use
torch.use_deterministic_algorithms(True)if torch has it instead oftorch.backends.cudnn.deterministic = True, as the later only makes convolutional deterministic.
Reactions are currently unavailable