scvi.model.base.PyroJitGuideWarmup.on_predict_epoch_end

PyroJitGuideWarmup.on_predict_epoch_end(trainer, pl_module, outputs)

Called when the predict epoch ends.

Return type

NoneNone