修改了自己模型的训练文件