Skip to content

Latest commit

 

History

History
7 lines (4 loc) · 670 Bytes

README.md

File metadata and controls

7 lines (4 loc) · 670 Bytes

text-rnn

text-rnn allows you to create modern neural network architectures which use modern techniques such as skip-embedding and attention weighting. It trains and generate text at the character-level. It also uses the CuDNN implementation when trained on GPUs which significantly improves training time when compared to the usual implementation of LSTMs.

You can configure whether to use bidirectional RNNs, the number of RNN layers, RNN size, input length, and size of the embedding layer.

If you would like to train using a free GPU check out this Colaboratory notebook.