#917 - Began the process of reloading existing agents into the session

This commit is contained in:
Chris McCarthy
2023-06-28 19:54:00 +01:00
parent 4866722911
commit b6d93ad33f
4 changed files with 82 additions and 18 deletions

View File

@@ -44,6 +44,8 @@ class SB3Agent(AgentSessionABC):
f"{self._training_config.agent_identifier}"
)
self.is_eval = False
def _setup(self):
super()._setup()
self._env = Primaite(
@@ -86,6 +88,7 @@ class SB3Agent(AgentSessionABC):
if not episodes:
episodes = self._training_config.num_episodes
self.is_eval = False
_LOGGER.info(f"Beginning learning for {episodes} episodes @"
f" {time_steps} time steps...")
for i in range(episodes):
@@ -108,6 +111,7 @@ class SB3Agent(AgentSessionABC):
if not episodes:
episodes = self._training_config.num_episodes
self._env.set_as_eval()
self.is_eval = True
if deterministic:
deterministic_str = "deterministic"
else: