projections.py 1.7 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

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)
X
xuwei06 已提交
29
    m6 += scaling_projection(m3)
30

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

with mixed_layer() as m7:
Q
qijun 已提交
35 36
    m7 += conv_operator(
        img=img, filter=flt, num_filters=64, num_channels=1, filter_size=3)
37
    m7 += conv_projection(img, filter_size=3, num_filters=64, num_channels=1)
Q
qijun 已提交
38

39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
with mixed_layer() as m8:
    m8 += conv_operator(
        img=img,
        filter=flt,
        num_filters=64,
        num_channels=1,
        filter_size=3,
        stride=2,
        padding=1,
        trans=True)
    m8 += conv_projection(
        img,
        filter_size=3,
        num_filters=64,
        num_channels=1,
        stride=2,
        padding=1,
        trans=True)
Q
qijun 已提交
57 58 59
end = mixed_layer(
    input=[
        full_matrix_projection(input=m5),
60 61
        trans_full_matrix_projection(input=m6),
        full_matrix_projection(input=m7), full_matrix_projection(input=m8)
Q
qijun 已提交
62 63 64 65
    ],
    size=100,
    layer_attr=ExtraAttr(
        drop_rate=0.5, error_clipping_threshold=40))
66 67

outputs(end)