scvi.model.base.PyroJitGuideWarmup.on_after_backward

PyroJitGuideWarmup.on_after_backward(trainer, pl_module)

Called after loss.backward() and before optimizers do anything.

Return type

NoneNone