浏览代码

update ts_batch_sampler for benchmark (#3762)

zhang-prog 7 月之前
父节点
当前提交
3408bd9813
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      paddlex/inference/common/batch_sampler/ts_batch_sampler.py

+ 2 - 2
paddlex/inference/common/batch_sampler/ts_batch_sampler.py

@@ -88,7 +88,7 @@ class TSBatchSampler(BaseBatchSampler):
                 batch.append(input, None)
                 if len(batch) == self.batch_size:
                     yield batch
-                    batch.reset()
+                    batch = Batch()
             elif isinstance(input, str):
                 file_path = (
                     self._download_from_url(input)
@@ -100,7 +100,7 @@ class TSBatchSampler(BaseBatchSampler):
                     batch.append(file_path, file_path)
                     if len(batch) == self.batch_size:
                         yield batch
-                        batch.reset()
+                        batch = Batch()
             else:
                 logging.warning(
                     f"Not supported input data type! Only `pd.DataFrame` and `str` are supported! So has been ignored: {input}."