scvi.train.SaveBestState#
- class scvi.train.SaveBestState(monitor='elbo_validation', mode='min', verbose=False, period=1)[source]#
Bases:
Callback
Save the best module state and restore into model.
- Parameters:
Examples
from scvi.train import Trainer from scvi.train import SaveBestState
Attributes table#
Methods table#
|
|
|
Called when the train ends. |
|
Called when the train begins. |
|
Called when the val epoch ends. |
Attributes#
Methods#
check_monitor_top
on_train_end
on_train_start
on_validation_epoch_end