I fixed and improved the Colab Notebook overall immensely, with new features (mount gdrive) and explanations (afaik)

[LtqxWYEG]

Talking about this notebook: https://colab.research.google.com/drive/1MEWKbm-driRNF8PrU7ogS5o3se-ePyPb?usp=sharing

There were issues when using symbols like "\" (for multiple phrases to be trained on) or any other symbol in the TEXT variable. I also added a couple more useful things and explanations as far as I understood the inner workings of everything. (Easily mounting your google drive, automatically saving images to a folder in it, checking what kind of GPU is used, made variables more descriptive etc)

My Colab Notebook: https://colab.research.google.com/drive/1zVHK4t3nXQTsu5AskOOOf3Mc9TnhltUO?usp=sharing

Thank you for providing all this!

Edit: Updated the notebook link since I have made many changes

[LtqxWYEG]

Added pull request: https://github.com/lucidrains/big-sleep/pull/121