deepke/example/ae/regular/conf/model/rnn.yaml

11 lines
235 B
YAML

model_name: rnn
type_rnn: 'LSTM' # [RNN, GRU, LSTM]
input_size: ??? # 使用 embedding 输出的结果,不需要指定
hidden_size: 150 # 必须为偶数
num_layers: 2
dropout: 0.3
bidirectional: True
last_layer_hn: True