zhang-prog пре 1 месец
родитељ
комит
4127933ab4
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      paddlex/inference/pipelines/_parallel.py

+ 1 - 1
paddlex/inference/pipelines/_parallel.py

@@ -44,7 +44,7 @@ class MultiDeviceSimpleInferenceExecutor(object):
                 input_future_pairs = []
                 for pipeline in self._pipelines:
                     try:
-                        input_batch = list(input_batches)[0]
+                        input_batch = next(input_batches)
                     except StopIteration:
                         out_of_data = True
                         break