wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-12 12:20:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

[rllib] Configure learner queue timeout (#5270)

Browse Source 
* configure learner queue timeout

* lint

* use config

* fix method args order, add unit test

* fix wrong param name
This commit is contained in:
Antoine Galataud
2019-07-25 21:18:05 -07:00
committed by Eric Liang
parent 6f682db99d
commit 827618254a
7 changed files with 50 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/optimizers/aso_learner.py
+5 -2
View File
@@ -26,14 +26,17 @@ class LearnerThread(threading.Thread):
"""
def __init__(self, local_worker, minibatch_buffer_size, num_sgd_iter,
learner_queue_size):
learner_queue_size, learner_queue_timeout):
threading.Thread.__init__(self)
self.learner_queue_size = WindowStat("size", 50)
self.local_worker = local_worker
self.inqueue = queue.Queue(maxsize=learner_queue_size)
self.outqueue = queue.Queue()
self.minibatch_buffer = MinibatchBuffer(
self.inqueue, minibatch_buffer_size, num_sgd_iter)
inqueue=self.inqueue,
size=minibatch_buffer_size,
timeout=learner_queue_timeout,
num_passes=num_sgd_iter)
self.queue_timer = TimerStat()
self.grad_timer = TimerStat()
self.load_timer = TimerStat()