github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-15 09:33:44 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix cpu fallback for extractor

Browse source
This commit is contained in:
iperov 2019-02-28 14:54:59 +04:00
parent 8f2e69ab27
commit 46f4d6c426
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
mainscripts/Extractor.py
View file

@ -193,6 +193,9 @@ class ExtractSubprocessor(Subprocessor):
io.progress_bar_close() io.progress_bar_close()
def get_devices_for_type (self, type, multi_gpu, cpu_only): def get_devices_for_type (self, type, multi_gpu, cpu_only):
if 'cpu' in nnlib.device.backend:
cpu_only = True
if not cpu_only and (type == 'rects' or type == 'landmarks'): if not cpu_only and (type == 'rects' or type == 'landmarks'):
if type == 'rects' and self.detector == 'mt' and nnlib.device.backend == "plaidML": if type == 'rects' and self.detector == 'mt' and nnlib.device.backend == "plaidML":
cpu_only = True cpu_only = True