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

#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

Browse Source
This commit is contained in:
Chris McCarthy
2023-07-06 13:56:12 +01:00
parent f92d2fb65d
commit 1e7f5b62f3
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/primaite/agents/rllib.py
View File

@@ -186,6 +186,7 @@ class RLlibAgent(AgentSessionABC):
checkpoint_dir = file checkpoint_dir = file
break break
# Zip the folder
shutil.make_archive( shutil.make_archive(
str(self._saved_agent_path).replace(".zip", ""), str(self._saved_agent_path).replace(".zip", ""),
"zip", "zip",