github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 13:02:15 -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

34
core/joblib/SubprocessorBase.py
View file

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