CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

decreased time of training initialization

Browse Source
This commit is contained in:
Colombo
2020-02-18 10:01:33 +04:00
parent 814da70577
commit 01376fd17c
4 changed files with 36 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
samplelib/SampleGeneratorFace.py
View File

@@ -60,7 +60,10 @@ class SampleGeneratorFace(SampleGeneratorBase):
if self.debug:
self.generators = [ThisThreadGenerator ( self.batch_func, (pickled_samples, index_host.create_cli(), ct_pickled_samples, ct_index_host.create_cli() if ct_index_host is not None else None) )]
else:
self.generators = [SubprocessGenerator ( self.batch_func, (pickled_samples, index_host.create_cli(), ct_pickled_samples, ct_index_host.create_cli() if ct_index_host is not None else None), start_now=True ) for i in range(self.generators_count) ]
self.generators = [SubprocessGenerator ( self.batch_func, (pickled_samples, index_host.create_cli(), ct_pickled_samples, ct_index_host.create_cli() if ct_index_host is not None else None), start_now=False ) \
for i in range(self.generators_count) ]
SubprocessGenerator.start_in_parallel( self.generators )
self.generator_counter = -1