1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00

remove unused steps parameter from GraphManager.train

This commit is contained in:
Zach Dwiel
2018-10-02 15:29:57 -04:00
committed by zach dwiel
parent 97f608ee5e
commit 7382a142bb
2 changed files with 3 additions and 3 deletions

View File

@@ -325,7 +325,7 @@ class GraphManager(object):
[environment.handle_episode_ended() for environment in self.environments]
def train(self, steps: TrainingSteps) -> None:
def train(self) -> None:
"""
Perform several training iterations for all the levels in the hierarchy
:param steps: number of training iterations to perform
@@ -426,7 +426,7 @@ class GraphManager(object):
# The actual steps being done on the environment are decided by the agents themselves.
# This is just an high-level controller.
self.act(EnvironmentSteps(1))
self.train(TrainingSteps(1))
self.train()
self.occasionally_save_checkpoint()
self.phase = RunPhase.UNDEFINED