Skip to content

S
seminar-breakout
You need to sign in or sign up before continuing.
Commit c81ea087 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

Join ZMQ processes after temination (#562)

parent ac02c62f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tensorpack/dataflow/prefetch.py
View file @ c81ea087
...@@ -108,6 +108,7 @@ class _MultiProcessZMQDataFlow(DataFlow): ...@@ -108,6 +108,7 @@ class _MultiProcessZMQDataFlow(DataFlow):
self.context.destroy(0) self.context.destroy(0)
for x in self._procs: for x in self._procs:
x.terminate() x.terminate()
x.join(5)
try: try:
print("{} successfully cleaned-up.".format(type(self).__name__)) print("{} successfully cleaned-up.".format(type(self).__name__))
except Exception: except Exception:
......
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