5.4 完整的训练代码