unable to train in colab

Created by: leo-bot