scvi.train.PyroTrainingPlanConfig

Contents

scvi.train.PyroTrainingPlanConfig#

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

Attributes table#

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

Methods table#

Attributes#

PyroTrainingPlanConfig.blocked: list | None = None#

PyroTrainingPlanConfig.loss_fn: Any | None = None#

PyroTrainingPlanConfig.n_epochs_kl_warmup: int | None = 400#

PyroTrainingPlanConfig.n_steps_kl_warmup: int | None = None#

PyroTrainingPlanConfig.optim: Any | None = None#

PyroTrainingPlanConfig.optim_kwargs: dict | None = None#

PyroTrainingPlanConfig.scale_elbo: float = 1.0#

Methods#

PyroTrainingPlanConfig.to_kwargs()[source]#

Return type:: dict[str, Any]