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): ...@@ -6,8 +6,8 @@ class EarlyStoppingVerbose(EarlyStopping):
The default EarlyStopping callback's verbose mode is too verbose. The default EarlyStopping callback's verbose mode is too verbose.
This class outputs a message only when it's getting ready to stop. This class outputs a message only when it's getting ready to stop.
""" """
def _evalute_stopping_criteria(self, *args): def _evalute_stopping_criteria(self, *args, **kwargs):
should_stop, reason = super()._evalute_stopping_criteria(*args) should_stop, reason = super()._evalute_stopping_criteria(*args, **kwargs)
if(should_stop): if(should_stop):
rank_zero_info(f"{reason}\n") 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