Make more friendly user outputs when training SB3

This commit is contained in:
Marek Wolan
2023-12-01 15:36:07 +00:00
parent 321d1f7219
commit eeedea2eff
10 changed files with 96 additions and 928 deletions

View File

@@ -62,6 +62,7 @@ class PrimaiteSession:
def start_session(self) -> None:
"""Commence the training/eval session."""
print("Staring Primaite Session")
self.mode = SessionMode.TRAIN
n_learn_episodes = self.training_options.n_learn_episodes
n_eval_episodes = self.training_options.n_eval_episodes