def METHOD_NAME(self, state_dict):
    """Load an LR scheduler state dict."""
    self.lr_scheduler.best = state_dict["best"]
    if "last_epoch" in state_dict:
        self.lr_scheduler.last_epoch = state_dict["last_epoch"]