4.2.4 创建扩散模型训练循环