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

extractor for AMD gpu: increase no response time

Browse source
This commit is contained in:
Colombo 2019-11-24 19:37:21 +04:00
parent 6b14741160
commit 1bfd65abe5
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
mainscripts/Extractor.py
View file

@ -307,9 +307,14 @@ class ExtractSubprocessor(Subprocessor):
self.result = [] self.result = []
self.devices = ExtractSubprocessor.get_devices_for_config(self.manual, self.type, multi_gpu, cpu_only) self.devices = ExtractSubprocessor.get_devices_for_config(self.manual, self.type, multi_gpu, cpu_only)
no_response_time_sec = 60 if not self.manual and not DEBUG else 999999
if self.manual or DEBUG:
no_response_time_sec = 999999
elif nnlib.device.backend == 'plaidML':
no_response_time_sec = 600
else:
no_response_time_sec = 60
super().__init__('Extractor', ExtractSubprocessor.Cli, no_response_time_sec, initialize_subprocesses_in_serial=(type != 'final')) super().__init__('Extractor', ExtractSubprocessor.Cli, no_response_time_sec, initialize_subprocesses_in_serial=(type != 'final'))
#override #override