src/transformers/optimization.py/get_adafactor_schedule
def get_adafactor_schedule(optimizer, initial_lr=0.0):
    """
    Get a proxy schedule for :class:`~transformers.optimization.Adafactor`

    Args:
        optimizer (:class:`~torch.optim.Optimizer`):
            The optimizer for which to schedule the learning rate.
        initial_lr (:obj:`float`, `optional`, defaults to 0.0):
            Initial lr

    Return:
        :class:`~transformers.optimization.Adafactor` proxy schedule object.


    """
    return AdafactorSchedule(optimizer, initial_lr)