def get_parameters():
params = tf.contrib.training.HParams(
# vocabulary
pad="</s>",
unk="UNK",
eos="</s>",
bos="</s>",
append_eos=False,
# model
rnn_cell="LegacyGRUCell",
embedding_size=620,
hidden_size=1000,
maxnum=2,
# regularization
dropout=0.2,
use_variational_dropout=False,
label_smoothing=0.1,
constant_batch_size=True,
batch_size=128,
max_length=60,
clip_grad_norm=5.0
)
return params
评论列表
文章目录