From dd848bc13a96038cb19e1d861dfc00abf4c49941 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Wed, 19 Apr 2023 20:11:49 +0200
Subject: [PATCH] force deterministic cudnn to help the tests
---
src/maggotuba/models/trainers.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/src/maggotuba/models/trainers.py b/src/maggotuba/models/trainers.py
index c0fae21..debb78c 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)
--
GitLab