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

Cleanup and refactoring (#171)

This commit is contained in:
Zach Dwiel
2019-01-15 03:04:53 -05:00
committed by Gal Leibovich
parent cd812b0d25
commit fedb4cbd7c
7 changed files with 45 additions and 33 deletions

View File

@@ -271,7 +271,7 @@ class LevelManager(EnvironmentInterface):
[agent.sync() for agent in self.agents.values()]
def should_train(self) -> bool:
return any([agent._should_train_helper() for agent in self.agents.values()])
return any([agent._should_update() for agent in self.agents.values()])
# TODO-remove - this is a temporary flow, used by the trainer worker, duplicated from observe() - need to create
# an external trainer flow reusing the existing flow and methods [e.g. observe(), step(), act()]