添加 训练函数