File size: 558 Bytes
ca7299e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | from pytorch_lightning.utilities import rank_zero_info
from pytorch_lightning.callbacks.early_stopping import EarlyStopping
class EarlyStoppingVerbose(EarlyStopping):
"""
The default EarlyStopping callback's verbose mode is too verbose.
This class outputs a message only when it's getting ready to stop.
"""
def _evalute_stopping_criteria(self, *args):
should_stop, reason = super()._evalute_stopping_criteria(*args)
if(should_stop):
rank_zero_info(f"{reason}\n")
return should_stop, reason
|