Method zero_grad

tools/train_utils/optimization/fastai_optim.py:154–156

Clear optimizer gradients.

(self)

        self.opt.step()

    def zero_grad(self) -> None:
        "Clear optimizer gradients."
        self.opt.zero_grad()

    # Passthrough to the inner opt.
    def __getattr__(self, k: str):