Commit 3fb643e9 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Merge branch 'main' of github.com:aqlaboratory/openfold into main

parents a70132f3 b9b8c0a5
......@@ -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")
......
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