This is a term project I did recently (2020) for my master's degree in information technology at Sabanci University.
When running the notebook file in Google Colab, make sure you have set Edit > Notebook settings to GPU, and have changed the directory variables accordingly.