scvi.train.LoudEarlyStopping#

class scvi.train.LoudEarlyStopping(**kwargs)[source]#

Bases: EarlyStopping

Wrapper of Pytorch Lightning EarlyStopping callback that prints the reason for stopping on teardown.

When the early stopping condition is met, the reason is saved to the callback instance, then printed on teardown. By printing on teardown, we do not interfere with the progress bar callback.

Attributes table#

Methods table#

teardown(_trainer, _pl_module[, stage])

Print the reason for stopping on teardown.

Attributes#

Methods#

LoudEarlyStopping.teardown(_trainer, _pl_module, stage=None)[source]#

Print the reason for stopping on teardown.

Return type:

None