未验证 提交 e24f53ad 编写于 作者: X Xiaoyao Xi 提交者: GitHub

Merge branch 'r0.3-api' into api-update

...@@ -56,6 +56,7 @@ def yield_pieces(data, distribute_strategy, batch_size): ...@@ -56,6 +56,7 @@ def yield_pieces(data, distribute_strategy, batch_size):
# print(len(temp)) # print(len(temp))
yield temp yield temp
def data_feeder(reader, postprocess_fn=None, prefetch_steps=2): def data_feeder(reader, postprocess_fn=None, prefetch_steps=2):
if postprocess_fn is None: if postprocess_fn is None:
def postprocess_fn(batch): def postprocess_fn(batch):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册