How to train your model on other datasets?

[JonasLi-19]

I think it will be nice if you offer a train.py script for the communities to utilize your work to get better explorations. Thanks!