fix S3FD for plaidML, fix S3FD warnings on Lambda layers.

2019-03-11 09:32:00 +04:00
parent 338a8fccc6
commit 893354b870
2 changed files with 1 additions and 1 deletions
@@ -213,7 +213,7 @@ class ExtractSubprocessor(Subprocessor):
            cpu_only = True
            
        if not cpu_only and (type == 'rects' or type == 'landmarks'):
-            if type == 'rects' and (self.detector == 'mt' or self.detector == 's3fd') and nnlib.device.backend == "plaidML":
+            if type == 'rects' and (self.detector == 'mt') and nnlib.device.backend == "plaidML":
                cpu_only = True
            else:
                if multi_gpu: