def _set_param_groups(self, groups):
        """A helper method to load param_groups from saved state_dict."""
        self._lr = groups[0]["lr"]
        self._beta1, self._beta2 = groups[0]["betas"]
        self._eps = groups[0]["eps"]
A helper method to load param_groups from saved state_dict.
python/dgl/optim/pytorch/sparse_optim.py