IIn this repository, we construct a character-level LSTM with PyTorch. The network will train character by character on some text, then generate new text character by character.
If you are using pip then you can install the required libraries by:
pip install -r requirements.txt
Conda users can make use of the env.yml file to activate
conda env create -f env.yml
conda activate env.yml
-
load the network
with open('rnn_x_epoch.net', 'rb') as f: checkpoint = torch.load(f) loaded = CharRNN(checkpoint['tokens'], n_hidden=checkpoint['n_hidden'], n_layers=checkpoint['n_layers']) loaded.load_state_dict(checkpoint['state_dict']) -
predict your text with
predict(net, ch, h, top_k=top_k)