scvi.model.base.PyroJitGuideWarmup.on_load_checkpoint

PyroJitGuideWarmup.on_load_checkpoint(trainer, pl_module, callback_state)

Called when loading a model checkpoint, use to reload state.

Parameters
trainer : TrainerTrainer

the current Trainer instance.

pl_module : LightningModuleLightningModule

the current LightningModule instance.

callback_state : {str: Any}Dict[str, Any]

the callback state returned by on_save_checkpoint.

Note

The on_load_checkpoint won’t be called with an undefined state. If your on_load_checkpoint hook behavior doesn’t rely on a state, you will still need to override on_save_checkpoint to return a dummy state.

Return type

NoneNone