Make saving step metadata optional

This commit is contained in:
Marek Wolan
2023-12-04 10:42:20 +00:00
parent 8ea9db2d34
commit a5c4f7797d
5 changed files with 17 additions and 2 deletions

View File

@@ -25,6 +25,8 @@ class SessionIOSettings(BaseModel):
"""Whether to save transactions, If true, the session path will have a transactions folder."""
save_tensorboard_logs: bool = False
"""Whether to save tensorboard logs. If true, the session path will have a tenorboard_logs folder."""
save_step_metadata: bool = False
"""Whether to save the RL agents' action, environment state, and other data at every single step."""
class SessionIO: