Skip to content

M
models

PaddlePaddle / models
大约 2 年 前同步成功

通知 232
Star 6828
Fork 2962
M
models
未验证 提交 35db3d17 编写于 作者: X Xing Wu 提交者： GitHub
浏览文件
操作

update .batch and .shuffle apis (#4620)

上级 eb35c88c
显示空白变更内容
内联 并排
Showing with 3 addition and 3 deletion
PaddleNLP/lexical_analysis/creator.py
浏览文件 @ 35db3d17
...@@ -105,15 +105,15 @@ def create_pyreader(args, ...@@ -105,15 +105,15 @@ def create_pyreader(args,
# create lac pyreader # create lac pyreader
if mode == 'train': if mode == 'train':
pyreader.set_sample_list_generator( pyreader.set_sample_list_generator(
paddle.batch( fluid.io.batch(
paddle.reader.shuffle( fluid.io.shuffle(
reader.file_reader(file_name), reader.file_reader(file_name),
buf_size=args.traindata_shuffle_buffer), buf_size=args.traindata_shuffle_buffer),
batch_size=args.batch_size / device_count), batch_size=args.batch_size / device_count),
places=place) places=place)
else: else:
pyreader.set_sample_list_generator( pyreader.set_sample_list_generator(
paddle.batch( fluid.io.batch(
reader.file_reader( reader.file_reader(
file_name, mode=mode), file_name, mode=mode),
batch_size=args.batch_size / device_count), batch_size=args.batch_size / device_count),
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或