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

compatibility fix

Browse source
This commit is contained in:
Colombo 2019-11-04 02:01:28 +04:00
parent a8a822fe68
commit 71a98e2ee4
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
mainscripts/Extractor.py
View file

@ -618,10 +618,8 @@ class ExtractSubprocessor(Subprocessor):
count = 1 count = 1
if not manual: if not manual:
if (type == 'rects-dlib' or type == 'rects-mt' ): if (type == 'rects-mt' ):
count = int (max (1, dev_vram / 2) ) count = int (max (1, dev_vram / 2) )
if type == 'rects-s3fd':
count = int (max (1, dev_vram / 5) )
if count == 1: if count == 1:
result += [ (idx, 'GPU', dev_name, dev_vram) ] result += [ (idx, 'GPU', dev_name, dev_vram) ]