def METHOD_NAME(self):
    config = super().METHOD_NAME()
    config.update(
        {
            "learning_rate": self._serialize_hyperparameter(
                self._learning_rate
            ),
            "beta_1": self.beta_1,
            "beta_2": self.beta_2,
        }
    )
    return config