这是一个通用的，基于pytorch的模型训练框架