scvi.model.base.PyroJitGuideWarmup.on_train_start

PyroJitGuideWarmup.on_train_start(trainer, pl_module)[source]

Way to warmup Pyro Guide in an automated way.

Also device agnostic.