def lr_schedule(num_epochs):
    """Some Learning rate schedule.

    Example:
        >>> # Halving the learning rate every epoch:
        >>> lambda epoch: 0.5 ** epoch
        >>> # A less aggressive decay:
        >>> lambda epoch: 0.9 ** epoch
        >>> # Constant learning rate (using init lr):
        >>> lambda epoch: 1.0

    Args:
        num_epochs (int): Maximum number of epochs to train.

    Returns:
        [callable]: Lambda function of the learning rate schedule.
    """
    return lambda epoch: 1.0