Train_eval
2022-07-14 14:46:30 3 举报
Pytorch训练验证流程示意图
作者其他创作
大纲/内容
优化器梯度清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
0 条评论
下一页
为你推荐
查看更多