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

optimized sample generator

Browse source
This commit is contained in:
Colombo 2020-01-05 11:53:31 +04:00
parent b5c234dac3
commit 21b25038ac
6 changed files with 201 additions and 160 deletions
Download patch file Download diff file Expand all files Collapse all files

2
utils/iter_utils.py
View file

@ -22,7 +22,7 @@ class ThisThreadGenerator(object):
return next(self.generator_func)
class SubprocessGenerator(object):
def __init__(self, generator_func, user_param=None, prefetch=2, start_now=False):
def __init__(self, generator_func, user_param=None, prefetch=3, start_now=False):
super().__init__()
self.prefetch = prefetch
self.generator_func = generator_func