Commit 39a6d0e6 authored by Christina Floristean's avatar Christina Floristean
Browse files

Merging in main branch

parents d8ee9c5f 84659c93
This diff is collapsed.
......@@ -6,8 +6,8 @@ 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)
def _evalute_stopping_criteria(self, *args, **kwargs):
should_stop, reason = super()._evalute_stopping_criteria(*args, **kwargs)
if(should_stop):
rank_zero_info(f"{reason}\n")
......
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment