From 1bfd65abe511d69dd989abb424a5159312a7c01a Mon Sep 17 00:00:00 2001 From: Colombo Date: Sun, 24 Nov 2019 19:37:21 +0400 Subject: [PATCH] extractor for AMD gpu: increase no response time --- mainscripts/Extractor.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/mainscripts/Extractor.py b/mainscripts/Extractor.py index 1b146bf..c023320 100644 --- a/mainscripts/Extractor.py +++ b/mainscripts/Extractor.py @@ -307,9 +307,14 @@ class ExtractSubprocessor(Subprocessor): self.result = [] 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')) #override