ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-12 07:35:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix process hangs at exit waiting for logger subprocess

Browse Source
This commit is contained in:
allegroai 2021-05-05 01:33:49 +03:00
parent e9bf34b13b
commit c9c4570569
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml/utilities/process/mp.py
View File

@ -289,16 +289,16 @@ class BackgroundMonitor(object):
execute_in_subprocess = task._report_subprocess_enabled execute_in_subprocess = task._report_subprocess_enabled
if not execute_in_subprocess: if not execute_in_subprocess:
for d in BackgroundMonitor._instances.get(id(task), []): for d in BackgroundMonitor._instances.get(id(task.id), []):
d._start() d._start()
elif not BackgroundMonitor._main_process: elif not BackgroundMonitor._main_process:
cls._parent_pid = os.getpid() cls._parent_pid = os.getpid()
cls._sub_process_started = SafeEvent() cls._sub_process_started = SafeEvent()
cls._sub_process_started.clear() cls._sub_process_started.clear()
# setup # setup
for d in BackgroundMonitor._instances.get(id(task), []): for d in BackgroundMonitor._instances.get(id(task.id), []):
d.set_subprocess_mode() d.set_subprocess_mode()
BackgroundMonitor._main_process = Process(target=cls._background_process_start, args=(id(task), )) BackgroundMonitor._main_process = Process(target=cls._background_process_start, args=(id(task.id), ))
BackgroundMonitor._main_process.daemon = True BackgroundMonitor._main_process.daemon = True
# Hack allow to create daemon subprocesses (even though python doesn't like it) # Hack allow to create daemon subprocesses (even though python doesn't like it)
un_daemonize = False un_daemonize = False