minimaxir / textgenrnn

Easily train your own text-generating neural network of any size and complexity on any text dataset with a few lines of code.
Other
4.94k stars 751 forks source link

TPU Support #79

Open minimaxir opened 5 years ago

minimaxir commented 5 years ago

Although TPUs are way overkill on a model of textgenrnn's size, Colaboratory offers to train using them for free, so support might be good.

It should just be a simple flag addition.

minimaxir commented 5 years ago

Tested it today. It'll probably have to wait until after the TensorFlow 2.0 merge because TPUs do not play nice with Keras's functional API. (not to mention I'm not 100% it'll run better since the current CuDNNLSTMs are pretty damn good.)