test_sequence_pooling.py 510 字节
Newer Older
1 2
from paddle.trainer_config_helpers import *

Q
qijun 已提交
3
settings(learning_rate=1e-4, batch_size=1000)
4 5 6

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

Q
qijun 已提交
7
POOL_TYPE = [MaxPooling, AvgPooling, SumPooling]
8

L
Luo Tao 已提交
9
AGG_LEVEL = [AggregateLevel.TO_SEQUENCE, AggregateLevel.TO_NO_SEQUENCE]
10 11 12 13 14 15 16

opts = []

for pt in POOL_TYPE:
    for al in AGG_LEVEL:
        opts.append(pooling_layer(input=din, agg_level=al, pooling_type=pt()))

Q
qijun 已提交
17 18 19
opts.append(
    pooling_layer(
        input=din, pooling_type=MaxPooling(output_max_index=True)))
20 21

outputs(opts)