scvi.model.base.PyroJitGuideWarmup.on_test_batch_start

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

Called when the test batch begins.

Return type

NoneNone