oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#2777: Add determinism to torch backends when seed set.

Browse Source
This commit is contained in:
Nick Todd
2024-08-05 15:08:31 +01:00
parent 7d7117e624
commit 966542c2ca
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/primaite/session/environment.py
View File

@@ -44,6 +44,10 @@ def set_random_seed(seed: int) -> Union[None, int]:
# if torch not installed don't set random seed. # if torch not installed don't set random seed.
if sys.modules["torch"]: if sys.modules["torch"]:
th.manual_seed(seed) th.manual_seed(seed)
th.backends.cudnn.deterministic = True
th.backends.cudnn.benchmark = False
return seed return seed