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

2
rl_coach/rollout_worker.py
View File

@@ -95,7 +95,7 @@ def rollout_worker(graph_manager, checkpoint_dir, data_store, num_workers):
break
if type(graph_manager.agent_params.algorithm.num_consecutive_playing_steps) == EnvironmentSteps:
graph_manager.act(EnvironmentSteps(num_steps=act_steps), wait_for_full_episode=graph_manager.agent_params.algorithm.act_for_full_episodes)
graph_manager.act(EnvironmentSteps(num_steps=act_steps), wait_for_full_episodes=graph_manager.agent_params.algorithm.act_for_full_episodes)
elif type(graph_manager.agent_params.algorithm.num_consecutive_playing_steps) == EnvironmentEpisodes:
graph_manager.act(EnvironmentEpisodes(num_steps=act_steps))