openfold.utils.callbacks

Classes

EarlyStoppingVerbose(monitor[, min_delta, ...])

The default EarlyStopping callback's verbose mode is too verbose.
class EarlyStoppingVerbose(monitor, min_delta=0.0, patience=3, verbose=False, mode='min', strict=True, check_finite=True, stopping_threshold=None, divergence_threshold=None, check_on_train_epoch_end=None, log_rank_zero_only=False)

Bases: EarlyStopping

The default EarlyStopping callback’s verbose mode is too verbose. This class outputs a message only when it’s getting ready to stop.

Parameters:

  • monitor (str)

  • min_delta (float)

  • patience (int)

  • verbose (bool)

  • mode (str)

  • strict (bool)

  • check_finite (bool)

  • stopping_threshold (float | None)

  • divergence_threshold (float | None)

  • check_on_train_epoch_end (bool | None)

  • log_rank_zero_only (bool)