I really beginner at this subject and I am trying to run the train here https://colab.research.google.com/drive/1Rb8OiF-AZ_Y3uFj1O2S0IyocFMhHoTCV?usp=sharing#scrollTo=7Xyt-G
