Skip to content

Commit 6d37e7b

Browse files
author
EnliteAI Bot
committed
RL-2039: Update MazeSeeding docs
(Issue RL-2039 - Add explicit seeding for training)
1 parent 6a99430 commit 6d37e7b

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

maze/core/utils/seeding.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,10 @@ class MazeSeeding:
4444
working on the gpu, however some torch modules will raise runtime errors, and the processing speed will be
4545
decreased. For more information on this topic please refer to:
4646
https://pytorch.org/docs/1.7.1/notes/randomness.html?highlight=reproducability
47-
47+
:param explicit_env_seeds: Explicit sequence of seeds used for seeding the environment (w.r.t. each episode).
48+
:param explicit_env_eval_seeds: Explicit sequence of seeds used for seeding the evaluation environment (w.r.t. each episode).
49+
:param explicit_agent_seeds: Explicit sequence of seeds used for seeding the agent (w.r.t. each episode).
50+
:param shuffle_seeds: Whether the explicit sequence of seeds should be shuffled (based on the env/agent _base_seed).
4851
"""
4952

5053
def __init__(self, env_seed: int, agent_seed: int, cudnn_determinism_flag: bool,

0 commit comments

Comments
 (0)