    def state_dict(self):
        """Return the LR scheduler state dict."""
        return {k: s.state_dict() for k, s in self.lr_schedulers.items()}
Return the LR scheduler state dict.
fairseq/optim/composite.py