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.