提交 db533188 编写于 作者: S shippingwang

refine

上级 662a3909
......@@ -206,7 +206,7 @@ def build(settings, mode='train'):
pipe, ['feed_image', 'feed_label'],
size=len(pipe),
dynamic_shape=True,
fill_last_batch=False,
fill_last_batch=True,
last_batch_padded=True)
file_list = os.path.join(file_root, 'train_list.txt')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册