Commit 771d895d by Lianmin Zheng Committed by Tianqi Chen

[AUTOTVM] Fix local executor (#1651)

The old queue size is too small. It will stall the executor due to race condition.
parent 0edf87e8
...@@ -133,7 +133,7 @@ class LocalExecutor(executor.Executor): ...@@ -133,7 +133,7 @@ class LocalExecutor(executor.Executor):
if not self.do_fork: if not self.do_fork:
return LocalFutureNoFork(func(*args, **kwargs)) return LocalFutureNoFork(func(*args, **kwargs))
queue = Queue(1) queue = Queue(2)
process = Process(target=timeout_monitor, process = Process(target=timeout_monitor,
args=(queue, self.timeout, func, args, kwargs)) args=(queue, self.timeout, func, args, kwargs))
process.start() process.start()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment