mnist_config.py 305 字节
Newer Older
Z
zhangjinchao01 已提交
1 2 3 4 5 6
from paddle.trainer_config_helpers import *

define_py_data_sources2(train_list='train.list',
                        test_list=None,
                        module='mnist_provider',
                        obj='process')
7 8
img = data_layer(name='pixel', size=784)
label = data_layer(name='label', size=10)