scvi.model.base.PyroJitGuideWarmup.on_epoch_start

PyroJitGuideWarmup.on_epoch_start(trainer, pl_module)

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

Return type

NoneNone