如何建立一个简单的模型训练工具包