scvi.train.LowLevelPyroTrainingPlanConfig

Contents

scvi.train.LowLevelPyroTrainingPlanConfig#

class scvi.train.LowLevelPyroTrainingPlanConfig(loss_fn=None, optim=None, optim_kwargs=None, n_steps_kl_warmup=None, n_epochs_kl_warmup=400, scale_elbo=1.0)[source]#: Config for LowLevelPyroTrainingPlan.

Attributes table#

`loss_fn`
`n_epochs_kl_warmup`
`n_steps_kl_warmup`
`optim`
`optim_kwargs`
`scale_elbo`

Methods table#

Attributes#

LowLevelPyroTrainingPlanConfig.loss_fn: Any | None = None#

LowLevelPyroTrainingPlanConfig.n_epochs_kl_warmup: int | None = 400#

LowLevelPyroTrainingPlanConfig.n_steps_kl_warmup: int | None = None#

LowLevelPyroTrainingPlanConfig.optim: Any | None = None#

LowLevelPyroTrainingPlanConfig.optim_kwargs: dict | None = None#

LowLevelPyroTrainingPlanConfig.scale_elbo: float = 1.0#

Methods#

LowLevelPyroTrainingPlanConfig.to_kwargs()[source]#

Return type:: dict[str, Any]