scvi.model.base.PyroSviTrainMixin

class scvi.model.base.PyroSviTrainMixin[source]

Mixin class for training Pyro models.

Training using minibatches and using full data (copies data to GPU only once).

Methods

train([max_epochs, use_gpu, train_size, …])

Train the model.