Method zero_grad

cdslib/core/optim/tf_optim.py:115–117

Zero the gradient buffer.

(self, set_to_none: bool = False)

        self._step = init_step

    def zero_grad(self, set_to_none: bool = False):
        """Zero the gradient buffer."""
        self.optimizer.zero_grad(set_to_none=set_to_none)

    def step(self, scaler=None):
        """Update parameters and learning rate."""