scvi.model.base.PyroJitGuideWarmup.on_batch_start

PyroJitGuideWarmup.on_batch_start(trainer, pl_module)

Called when the training batch begins.

Return type

NoneNone