优化器梯度清0用当前批次训练
设置模型状态train/eval
更新权重
span class=\"equation-text\" contenteditable=\"false\" data-index=\"0\" data-equation=\
验证/测试的流程基本与训练过程一致,不同点在于:1. 需要预先设置torch.no_grad,以及将model调至eval模式2. 不需要将优化器的梯度置零3. 不需要将loss反向回传到网络4. 不需要更新optimizer
循环读取数据传到GPU