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

Handle both Environment Steps and Episodes on the subscriber side. (#99)

Browse Source
This commit is contained in:
Balaji Subramaniam
2018-11-15 14:42:21 -08:00
committed by Scott Leishman
parent 3358e04a6a
commit 101c55d37d
4 changed files with 23 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rl_coach/graph_managers/graph_manager.py
View File

@@ -659,9 +659,9 @@ class GraphManager(object):
self.handle_episode_ended()
self.reset_required = True
def fetch_from_worker(self, num_steps=0):
def fetch_from_worker(self, num_consecutive_playing_steps=None):
if hasattr(self, 'memory_backend'):
for transition in self.memory_backend.fetch(num_steps):
for transition in self.memory_backend.fetch(num_consecutive_playing_steps):
self.emulate_act_on_trainer(EnvironmentSteps(1), transition)
def setup_memory_backend(self) -> None: