[tune] Fix GPU thread leakage

2026-06-29 08:14:56 +08:00 · 2020-10-01 17:07:13 -07:00
parent 0d5b09f426
commit f60d77d78e
1 changed files with 3 additions and 0 deletions
@@ -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()