trainer_config.py 420 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13
from paddle.trainer_config_helpers import *

WORD_DIM = 3000

sentence = data_layer(name='sentence', size=WORD_DIM)
sentence_embedding = embedding_layer(
    input=sentence,
    size=64,
    param_attr=ParameterAttribute(
        initial_max=1.0, initial_min=0.5))
lstm = simple_lstm(input=sentence_embedding, size=64)
lstm_last = last_seq(input=lstm)
outputs(fc_layer(input=lstm_last, size=2, act=SoftmaxActivation()))