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

bug-fix for dumping movies (+ small refactoring and rename 'VideoDumpMethod -> 'VideoDumpFilter')

This commit is contained in:
Gal Leibovich
2018-10-21 17:29:10 +03:00
parent 364168490f
commit 5a8da90d32
4 changed files with 98 additions and 90 deletions

View File

@@ -325,8 +325,7 @@ class GraphManager(object):
"""
self.total_steps_counters[self.phase][EnvironmentEpisodes] += 1
# TODO: we should disentangle ending the episode from resetting the internal state
# self.reset_internal_state()
[environment.handle_episode_ended() for environment in self.environments]
def train(self, steps: TrainingSteps) -> None:
"""