Training not working run on colab 