From 795bf394954e61b30621a5524e1be67b69db506b Mon Sep 17 00:00:00 2001 From: Jan Date: Thu, 9 Dec 2021 16:01:57 +0100 Subject: [PATCH] cap cpu during filtering --- samplelib/SampleGeneratorFaceXSeg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/samplelib/SampleGeneratorFaceXSeg.py b/samplelib/SampleGeneratorFaceXSeg.py index 334213d..240f721 100644 --- a/samplelib/SampleGeneratorFaceXSeg.py +++ b/samplelib/SampleGeneratorFaceXSeg.py @@ -209,7 +209,7 @@ class SegmentedSampleFilterSubprocessor(Subprocessor): #override 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} #override