Added A example Training Script which can run on Google Colab on a single GPU