有两种方式直接把模型的参数梯度设成0:
model.zero_grad()
optimizer.zero_grad() # 当optimizer=optim.Optimizer(model.parameters())时,两者等效model.zero_grad()
optimiz有两种方式直接把模型的参数梯度设成0:
model.zero_grad()
optimizer.zero_grad() # 当optimizer=optim.Optimizer(model.parameters())时,两者等效model.zero_grad()
optimiz