Created by: Mariya Shavshyshvili
email: mariyasha888@gmail.com
Text-generating Neural Network, trained on Fairytales by The Brothers Grimm and Alice's Adventures in Wonderland by Lewis Carroll.
GPU connection is highly recomended for taining, as it may take over 10 hours to train on CPU.
Text files were generated from:
Fairytales: https://www.gutenberg.org/files/2591/2591-h/2591-h.htm
Alice: http://www.gutenberg.org/files/11/11-0.txt
IMPORTANT NOTES
Please replace the following variables in the notebook with the .txt files in the repository,
as this notebook links to my own Google Drive.
fairytales_url : ensure it links to FairytalesByTheBrothersGrimm.txt
alice_url : ensure it links to AlicesAdvanturesInWonderland.txt
checkpoint_url : any location to save model's checkpoint after training
Got lots of help with the word embeddings from Pytorch:
https://pytorch.org/tutorials/beginner/nlp/word_embeddings_tutorial.html#exercise-computing-word-embeddings-continuous-bag-of-words
Extra help regarding Google Colab's GPU connection:
https://www.kdnuggets.com/2018/02/fast-ai-lesson-1-google-colab-free-gpu.html
MariyaSha/StoryTeller
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|