model.zero_grad(),optimizer.zero_grad(),Variable.grad.data.zero_()
将模型参数梯度设置为0
model.zero_grad()
optimizer.zero_grad()#当optimizer=optim.Optimizer(model.parameters())时,两者等效
单一变量Variable梯度设置为0
Variable.grad.data.zero_()
原创
2020-11-10 10:27:33 ·
421 阅读 ·
0 评论