scvi.model.base.PyroJitGuideWarmup.on_predict_batch_start

PyroJitGuideWarmup.on_predict_batch_start(trainer, pl_module, batch, batch_idx, dataloader_idx)

Called when the predict batch begins.

Return type

NoneNone