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
-
Notifications
You must be signed in to change notification settings - Fork 4
MariyaSha/StoryTeller
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Text-generating Neural Network, trained with GPU in Google Colab using Pytorch and word embeddings
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published