5大典型模型测试单机训练速度超对标框架,飞桨如何做到?( 五 )

defdata_reader(width, height):

defreader:

whileTrue:

yieldnp.random.uniform( -1, 1, size=width*height),

np.random.randint( 0, 10)

returnreader

train_data_reader = data_reader( 32, 32)

飞桨提供了两种方式从Data Reader中读取数据:同步数据读取和异步数据读取。

2.1 同步数据读取

同步数据读取是一种简单并且直观的数据准备方式,代码示例如下:

Image = paddle.layer.data( "image", ...)

label = paddle.layer.data( "label", ...)

# 模型定义

# ……

prediction = fluid.layers.fc(input= image, size=10)

推荐阅读