gryf/coach
1
0 Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-10-13 18:22:29 +02: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 authored and Zach Dwiel committed 2019-04-09 12:14:27 -04:00
1 parent 06de3b0f07
commit dddaefb210
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
rl_coach/rollout_worker.py
+1 -1
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