diff --git a/src/maggotuba/models/trainers.py b/src/maggotuba/models/trainers.py index c0fae21cb459d1c2ac2211617d8199b6c2e23a28..debb78c8c77dbbe767ebc364fd4ca3073faca7be 100644 --- a/src/maggotuba/models/trainers.py +++ b/src/maggotuba/models/trainers.py @@ -259,7 +259,8 @@ def enforce_reproducibility(generator=None): seed = generator.initial_seed() # see https://pytorch.org/docs/1.13/notes/randomness.html torch.use_deterministic_algorithms(True) - # torch.backends.cudnn.deterministic = True + if torch.cuda.is_available: + torch.backends.cudnn.deterministic = True torch.manual_seed(seed) seed = seed % 2**32 np.random.seed(seed)