github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 13:32:09 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix

Browse source
This commit is contained in:
Colombo 2020-02-29 13:31:36 +04:00
parent e9c0f5fb9f
commit 7180076ecf
1 changed files with 21 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

8
core/joblib/SubprocessorBase.py
View file

@ -29,6 +29,8 @@ class Subprocessor(object):
def kill(self): def kill(self):
self.p.terminate() self.p.terminate()
self.p.join() self.p.join()
self.s2c.close()
self.c2s.close()
#overridable optional #overridable optional
def on_initialize(self, client_dict): def on_initialize(self, client_dict):
@ -77,7 +79,12 @@ class Subprocessor(object):
time.sleep(0.001) time.sleep(0.001)
self.on_finalize() self.on_finalize()
c2s.put ( {'op': 'finalized'} ) c2s.put ( {'op': 'finalized'} )
self.s2c.close()
self.c2s.close()
return return
except Subprocessor.SilenceException as e: except Subprocessor.SilenceException as e:
pass pass
@ -281,6 +288,7 @@ class Subprocessor(object):
while True: while True:
for cli in self.clis[:]: for cli in self.clis[:]:
if cli.state != 2:
terminate_it = False terminate_it = False
while not cli.c2s.empty(): while not cli.c2s.empty():
obj = cli.c2s.get() obj = cli.c2s.get()