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

cap cpu during filtering

Browse source
This commit is contained in:
Jan 2021-12-09 16:01:57 +01:00
commit 795bf39495
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
samplelib/SampleGeneratorFaceXSeg.py
View file

@ -209,7 +209,7 @@ class SegmentedSampleFilterSubprocessor(Subprocessor):
#override #override
def process_info_generator(self): def process_info_generator(self):
for i in range(multiprocessing.cpu_count()): for i in range(min(multiprocessing.cpu_count(),8)):
yield 'CPU%d' % (i), {}, {'samples':self.samples, 'count_xseg_mask':self.count_xseg_mask} yield 'CPU%d' % (i), {}, {'samples':self.samples, 'count_xseg_mask':self.count_xseg_mask}
#override #override