Commit 6f02650d authored by Yuxin Wu's avatar Yuxin Wu

[GPUUtilizationTracker] terminate the worker process at exit (instead of waiting it) (#1134)

parent 6e0b509b
......@@ -84,7 +84,7 @@ class GPUUtilizationTracker(Callback):
def _after_train(self):
self._stop_evt.set()
self._evt.set()
self._proc.join()
self._proc.terminate()
def worker(self, evt, rst_queue, stop_evt):
while True:
......
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