[P] Python package to easily retrain OpenAI’s GPT-2 text-generating model on new texts + Colaboratory Notebook to use it w/ GPU for free
Hi all! I just open-sourced a Python package on GitHub that lets you retrain the smaller GPT-2 model on your own text with minimal code! (and without fussing around with the CLI like the original repo)
I have also made a Colaboratory Notebook which handles both training w/ a GPU for free and file I/O to the notebook (which with GPT-2 is a tad tricker).
Let me know if you have any questions! I plan on releasing more demos soon!