Skip to content

Latest commit

 

History

History
34 lines (15 loc) · 598 Bytes

README.md

File metadata and controls

34 lines (15 loc) · 598 Bytes

new-LSTM-Cell

论文实现 by - pytorch

这是基于pytorch实现的该论文模型

主要在于改变了LSTM内部实现

分为:

LSTMCell_AddC -> encoder部分使用的LSTM

LSTMCell_AddC_decoder -> decoder部分使用的LSTM

其中:

lstm-lstm-pytorch-evaluate-embed.py是模型部分

rnn.py是根据需要修改了pytorch的内部实现

使用:

将pytorch下的nn/_function/rnn.py替换即可

...-GPU 是支持GPU运行的版本

...-embed 是本地运行版本