search

antihutka / pytorch-rnn

torch-rnn rewrite in pytorch
MIT License
9 stars 4 forks source link

LSTM #1

Open ruvsv opened 4 years ago

ruvsv commented 4 years ago

It would not be bad if LSTM model was supported.

antihutka commented 4 years ago

I added support for PyTorch's LSTM implementation. Unfortunately it's not compatible with torch-rnn's LSTM implementation. I did some quick tests and it seems to work well. Let me know if there are any problems with it.

ruvsv commented 4 years ago

O! Thank You. I try soon this new method on my corpus and compare the results with the torch-rnn.