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/training_worker.py
View File

@@ -36,7 +36,7 @@ def training_worker(graph_manager, checkpoint_dir):
while(steps < graph_manager.improve_steps.num_steps):
graph_manager.phase = core_types.RunPhase.TRAIN
graph_manager.fetch_from_worker(num_steps=graph_manager.agent_params.algorithm.num_consecutive_playing_steps.num_steps)
graph_manager.fetch_from_worker(graph_manager.agent_params.algorithm.num_consecutive_playing_steps)
graph_manager.phase = core_types.RunPhase.UNDEFINED
if graph_manager.should_train():