scvi.model.base.PyroJitGuideWarmup.on_train_batch_end

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

Called when the train batch ends.

Return type

NoneNone