fixed bug in rollout worker where total number of improved steps are not taken
This commit is contained in:
1 parent
06de3b0f07
commit
dddaefb210
1 file changed
+1
-1
@@ -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
|
||||
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):
|
||||
break
|
||||
|
||||
Reference in new issue
Block a user