模型训练框架优化

1