scvi.train.Trainer.fit

Trainer.fit(*args, **kwargs)[source]

Runs the full optimization routine.

Parameters
model

Model to fit.

train_dataloader

Either a single PyTorch DataLoader or a collection of these (list, dict, nested lists and dicts). In the case of multiple dataloaders, please see this page

val_dataloaders

Either a single Pytorch Dataloader or a list of them, specifying validation samples. If the model has a predefined val_dataloaders method this will be skipped

datamodule

An instance of LightningDataModule.