aleksimyllyoja / char-rnn

Character-level Recurrent Neural Network model

Char-rnn generation

Character-level language model implementation of Karpathy's char-rnn (http://karpathy.github.io/2015/05/21/rnn-effectiveness/)

• Current version is written with Pytorch and GRU shell

• chainer/ contains old version writen with Chainer and LSTM shell

Requirements

• Python >= 3.4

• Pytorch

Train

python train.py --epoch 5000 --frequency 100 --gpu 0 --source harry.txt

Model is saved in save/char-rnn-gru.pt

Generate

python generate.py --model save/char-rnn-gru.pt --prime "Harry Potter" --len 2000