scvi.model.base.PyroJitGuideWarmup.on_test_batch_end

PyroJitGuideWarmup.on_test_batch_end(trainer, pl_module, outputs, batch, batch_idx, dataloader_idx)

Called when the test batch ends.

Return type

NoneNone