scvi.model.base.PyroJitGuideWarmup.on_fit_end

PyroJitGuideWarmup.on_fit_end(trainer, pl_module)

Called when fit ends

Return type

NoneNone