[tune] Fix GPU thread leakage

This commit is contained in:
Richard Liaw authored and GitHub committed 2020-10-01 17:07:13 -07:00
1 parent 0d5b09f426
commit f60d77d78e
1 file changed
+3
+3
View File
@@ -637,6 +637,9 @@ class Trainable:
"""
self._result_logger.flush()
self._result_logger.close()
if self._monitor.is_alive():
self._monitor.stop()
self._monitor.join()
self.cleanup()
self._close_logfiles()