def METHOD_NAME(self) -> Dict[str, Any]:
    """Configure the model's optimizer(s)."""
    optimizer = self._optimizer_class(
        self.parameters(), **self._optimizer_kwargs
    )
    config = {
        "optimizer": optimizer,
    }
    if self._scheduler_class is not None:
        scheduler = self._scheduler_class(
            optimizer, **self._scheduler_kwargs
        )
        config.update(
            {
                "lr_scheduler": {
                    "scheduler": scheduler,
                    **self._scheduler_config,
                },
            }
        )
    return config