- class scvi.train.LoudEarlyStopping(**kwargs)[source]#
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.
Print the reason for stopping on teardown.