4.5 基于PyTorch的模型训练实现