projections.py 1.1 KB
Newer Older
1 2 3 4 5
'''
Test mixed layer, projections and operators.
'''
from paddle.trainer_config_helpers import *

Q
qijun 已提交
6
settings(batch_size=1000, learning_rate=1e-4)
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29

din = data_layer(name='test', size=100)

din = embedding_layer(input=din, size=256)

with mixed_layer(size=100) as m1:
    m1 += full_matrix_projection(input=din)

with mixed_layer(size=100) as m2:
    m2 += table_projection(input=m1)

with mixed_layer(size=100) as m3:
    m3 += identity_projection(input=m2)

with mixed_layer(size=100) as m4:
    m4 += dotmul_projection(input=m3)

with mixed_layer() as m5:
    m5 += context_projection(input=m4, context_len=3)

with mixed_layer() as m6:
    m6 += dotmul_operator(a=m3, b=m4)

Q
qijun 已提交
30 31
img = data_layer(name='img', size=32 * 32)
flt = data_layer(name='filter', size=3 * 3 * 1 * 64)
32 33

with mixed_layer() as m7:
Q
qijun 已提交
34 35 36 37 38 39 40 41 42 43 44
    m7 += conv_operator(
        img=img, filter=flt, num_filters=64, num_channels=1, filter_size=3)

end = mixed_layer(
    input=[
        full_matrix_projection(input=m5),
        trans_full_matrix_projection(input=m6), full_matrix_projection(input=m7)
    ],
    size=100,
    layer_attr=ExtraAttr(
        drop_rate=0.5, error_clipping_threshold=40))
45 46

outputs(end)