1、 Tiny_LSTM自定义实现了 LSTM Cell 及其FPTT、BPTT 和Weight Update,可以快速进行模型搭建,快速运算。
2、 基于该框架快速计算模型精度如下:
Mnist Accuracy : 96%
Net 结构 : LSTM + Softmax_Cross_Entropy (seqLength = 28, inputSize = 28, hiddenSize = 128 )
LSTM Cell 包括 Update Cell, Compute Cell, Forget Cell, Output Cell。
3、 水平有限,代码有不完善的地方请在ISSUE批评指正!