def lr_schedule(epoch):
    """
    Learning Rate Scheduler for Adam
    """
    lr = 0.0001
    if epoch > 7:
        lr = 0.00008
    elif epoch > 9:
        lr = 0.00005
    elif epoch > 11:
        lr = 0.00001
    
    return lr