参数

输入尺寸(input-size)

128

序列长度(sequence-length)

60

批尺寸(batch-size)

16

隐藏层(hidden-size)

32

输出层(output-size)

2

模型层数(num-layer)

1

随机失活率(dropout)

0.5

优化器(optimizer)

SGD

损失函数

Cross Entropy Loss