new-LSTM-Cell icon indicating copy to clipboard operation
new-LSTM-Cell copied to clipboard

论文实现 by - pytorch

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 是本地运行版本