gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 11:40:18 +01:00
Code Issues Projects Releases Wiki Activity

very minor cleanup

Browse Source
This commit is contained in:
Zach Dwiel
2018-09-11 13:42:50 -04:00
parent a16d724963
commit 673911ff7f
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
rl_coach/graph_managers/graph_manager.py
View File

@@ -288,8 +288,7 @@ class GraphManager(object):
count_end = self.total_steps_counters[RunPhase.TRAIN][TrainingSteps] + steps.num_steps
while self.total_steps_counters[RunPhase.TRAIN][TrainingSteps] < count_end:
self.total_steps_counters[RunPhase.TRAIN][TrainingSteps] += 1
losses = [manager.train() for manager in self.level_managers]
# self.loss.add_sample(loss)
[manager.train() for manager in self.level_managers]
def reset_internal_state(self, force_environment_reset=False) -> None:
"""