nip.utils.maths._zero_grad

Contents

nip.utils.maths._zero_grad#

nip.utils.maths._zero_grad(params: dict[str, Tensor | Parameter])[source]#

Zero the gradients of the parameters in a dictionary.

Parameters:

params (dict[str, Tensor | Parameter]) – The dictionary of parameters.