lightning/pytorch_lightning/callbacks/lr_logger.py

10 lines
416 B
Python

from pytorch_lightning.callbacks.lr_monitor import LearningRateMonitor
from pytorch_lightning.utilities import rank_zero_warn
class LearningRateLogger(LearningRateMonitor):
def __init__(self, *args, **kwargs):
rank_zero_warn("`LearningRateLogger` is now `LearningRateMonitor`"
" and this will be removed in v0.11.0", DeprecationWarning)
super().__init__(*args, **kwargs)