I tried to train the custom object according to https://www.youtube.com/watch?v=mmj3nxGT2YQ&t=1747s. In the training part when i run this code on my colab