scvi.model.base.PyroJitGuideWarmup.teardown

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

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

Return type

NoneNone