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

Don't pass multiprocess object on pickle (#261)

Browse source
This commit is contained in:
Luke Barr 2019-05-14 02:30:23 -05:00 • committed by iperov
parent d6e8dde481
commit ba4e377d13
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
utils/iter_utils.py
View file

@ -48,6 +48,11 @@ class SubprocessGenerator(object):
def __iter__(self): def __iter__(self):
return self return self
def __getstate__(self):
self_dict = self.__dict__.copy()
del self_dict['p']
return self_dict
def __next__(self): def __next__(self):
if self.p == None: if self.p == None:
user_param = self.user_param user_param = self.user_param