def METHOD_NAME(self, epoch):
    manual_keys = [k for k in self.epoch2lr if k <= epoch]
    if manual_keys:
        manual_lr = self.epoch2lr[max(manual_keys)]
    else:
        logger.warning(
            "@@@ epoch={} does not exist in manual lr input. epoch2lr={}...".format(
                epoch,
                list(self.epoch2lr.items())[
                    : min(10, len(self.epoch2lr.keys()) - 1)
                ],
            )
        )
        manual_lr = self.optimizer.get_lr()
    return manual_lr