Train_eval

优化器梯度清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