test_maxout.py 1.4 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
from paddle.trainer_config_helpers import *

settings(
    batch_size=1000,
    learning_rate=1e-5
)

data = data_layer(name='data', size=2304)

conv = img_conv_layer(input=data,
                      filter_size = 3,
                      num_channels=1,
                      num_filters=16,
                      padding=1,
                      act=LinearActivation(),
                      bias_attr=True)

maxout = maxout_layer(input=conv,
                      num_channels=16,
                      groups=2)

pool = img_pool_layer(input=maxout,
                      num_channels=8,
                      pool_size=2,
                      stride=2,
                      pool_type=MaxPooling())

28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
conv2 = img_conv_layer(input=pool,
                       filter_size = 3,
                       num_channels=32,
                       num_filters=128,
                       padding=1,
                       act=LinearActivation(),
                       bias_attr=True)

maxout2 = maxout_layer(input=conv,
                       num_channels=128,
                       groups=4)

block = block_expand_layer(input=maxout,
                           num_channels=32,
                           stride_x=1,
                           stride_y=1,
                           block_x=1,
                           block_y=6)

fc = fc_layer(input=block, size=384, bias_attr=False)
48 49

outputs(fc)