#1963 - Made RLlib and SB3 agents save at the end of each learning session by default using a common file naming format. Also now agents only checkpoint every n and not on the final episode

2023-07-06 13:56:12 +01:00
parent 8faf9d70a0
commit 159d47fd6c
1 changed files with 1 additions and 0 deletions
--- a/src/primaite/agents/rllib.py
+++ b/src/primaite/agents/rllib.py
@@ -186,6 +186,7 @@ class RLlibAgent(AgentSessionABC):
            checkpoint_dir = file
            break

+        # Zip the folder
        shutil.make_archive(
            str(self._saved_agent_path).replace(".zip", ""),
            "zip",