|
|
|
@ -15,12 +15,16 @@
|
|
|
|
|
from paddle.trainer_config_helpers import *
|
|
|
|
|
|
|
|
|
|
################################### Data Configuration ###################################
|
|
|
|
|
TrainData(ProtoData(files = "trainer/tests/mnist.list"))
|
|
|
|
|
TrainData(SimpleData(
|
|
|
|
|
files = "trainer/tests/sample_filelist.txt",
|
|
|
|
|
feat_dim = 3,
|
|
|
|
|
context_len = 0,
|
|
|
|
|
buffer_capacity = 1000000))
|
|
|
|
|
################################### Algorithm Configuration ###################################
|
|
|
|
|
settings(batch_size = 1000,
|
|
|
|
|
learning_method = MomentumOptimizer(momentum=0.5, sparse=False))
|
|
|
|
|
################################### Network Configuration ###################################
|
|
|
|
|
data = data_layer(name ="input", size=784)
|
|
|
|
|
data = data_layer(name ="input", size=3)
|
|
|
|
|
|
|
|
|
|
fc1 = fc_layer(input=data, size=800,
|
|
|
|
|
bias_attr=True,
|
|
|
|
|