#917 - Synced with dev and added better logging

2023-06-28 12:01:01 +01:00
parent edab1a393d
commit a9ebfd7917
12 changed files with 170 additions and 87 deletions
--- a/src/primaite/agents/rllib.py
+++ b/src/primaite/agents/rllib.py
@@ -152,7 +152,8 @@ class RLlibAgent(AgentSessionABC):

        if not episodes:
            episodes = self._training_config.num_episodes
-
+        _LOGGER.info(f"Beginning learning for {episodes} episodes @"
+                     f" {time_steps} time steps...")
        for i in range(episodes):
            self._current_result = self._agent.train()
            self._save_checkpoint()