scvi.model.base.PyroJitGuideWarmup.setup

PyroJitGuideWarmup.setup(trainer, pl_module, stage=None)

Called when fit, validate, test, predict, or tune begins

Return type

NoneNone