Method zero_grad

python/dgl/optim/pytorch/sparse_optim.py:451–453

clean grad cache

(self)

        """

    def zero_grad(self):
        """clean grad cache"""
        self._clean_grad = True

    def state_dict(self, **kwargs):  # pylint: disable=unused-argument
        """Return a copy of the whole optimizer states stored in CPU memory.