scvi.model.base.PyroJitGuideWarmup.on_epoch_end

PyroJitGuideWarmup.on_epoch_end(trainer, pl_module)

Called when either of train/val/test epoch ends.

Return type

NoneNone