mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 13:02:15 -07:00
Don't pass multiprocess object on pickle (#261)
This commit is contained in:
parent
d6e8dde481
commit
ba4e377d13
1 changed files with 5 additions and 0 deletions
|
@ -48,6 +48,11 @@ class SubprocessGenerator(object):
|
|||
def __iter__(self):
|
||||
return self
|
||||
|
||||
def __getstate__(self):
|
||||
self_dict = self.__dict__.copy()
|
||||
del self_dict['p']
|
||||
return self_dict
|
||||
|
||||
def __next__(self):
|
||||
if self.p == None:
|
||||
user_param = self.user_param
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue