oneflow.nn.Module.zero_grad

Module.zero_grad(set_to_none=False)

Sets gradients of all model parameters to zero. See similar function under oneflow.optim.Optimizer for more context.

Parameters

set_to_none (bool) – instead of setting to zero, set the grads to None. See oneflow.optim.Optimizer.zero_grad() for details.