素质三连 optimizer.zero_grad() # 清空过往梯度； loss.backward(retain_graph=True) # 反向传播，计算当前梯度； optimizer.step() # 根据梯度更新网络参数
本文介绍使用PyTorch进行模型训练的基本步骤：清空过往梯度、反向传播计算当前梯度并保留计算图、以及根据梯度更新网络参数。
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
