High level wrapper to train pyTorch models on common datasets - View it on GitHub