-
-
Notifications
You must be signed in to change notification settings - Fork 654
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.