scvi.model.base.PyroJitGuideWarmup.on_pretrain_routine_end

PyroJitGuideWarmup.on_pretrain_routine_end(trainer, pl_module)

Called when the pretrain routine ends.

Return type

NoneNone