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

fix for amd

Browse source
This commit is contained in:
Colombo 2019-10-24 21:59:13 +04:00
parent 0448a461f5
commit 75eeef0a96
2 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
mainscripts/Extractor.py
View file

@ -76,7 +76,7 @@ class ExtractSubprocessor(Subprocessor):
self.e = facelib.DLIBExtractor(nnlib.dlib)
elif self.type == 'rects-s3fd':
nnlib.import_all (device_config)
self.e = facelib.S3FDExtractor()
self.e = facelib.S3FDExtractor(do_dummy_predict=True)
else:
raise ValueError ("Wrong type.")
@ -88,7 +88,7 @@ class ExtractSubprocessor(Subprocessor):
self.e = facelib.FANExtractor()
self.e.__enter__()
if self.device_vram >= 2:
self.second_pass_e = facelib.S3FDExtractor()
self.second_pass_e = facelib.S3FDExtractor(do_dummy_predict=False)
self.second_pass_e.__enter__()
else:
self.second_pass_e = None