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

fixed bug in rollout worker where total number of improved steps are not taken

Browse Source
This commit is contained in:
zach dwiel
2019-04-04 14:55:31 -04:00
committed by Zach Dwiel
parent 06de3b0f07
commit dddaefb210
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/rollout_worker.py
View File

@@ -79,7 +79,7 @@ def rollout_worker(graph_manager, data_store, num_workers, task_parameters):
# this worker should play a fraction of the total playing steps per rollout # this worker should play a fraction of the total playing steps per rollout
act_steps = math.ceil((graph_manager.agent_params.algorithm.num_consecutive_playing_steps.num_steps)/num_workers) act_steps = math.ceil((graph_manager.agent_params.algorithm.num_consecutive_playing_steps.num_steps)/num_workers)
for i in range(int(graph_manager.improve_steps.num_steps/act_steps)): for i in range(math.ceil(graph_manager.improve_steps.num_steps/act_steps)):
if should_stop(checkpoint_dir): if should_stop(checkpoint_dir):
break break