scvi.external.gimvi._task.GIMVITrainingPlan

scvi.external.gimvi._task.GIMVITrainingPlan#

class scvi.external.gimvi._task.GIMVITrainingPlan(*args, **kwargs)[source]#

Bases: AdversarialTrainingPlan

gimVI training plan.

Attributes table#

training

Methods table#

on_validation_epoch_end()

Aggregate validation step information.

training_step(batch, batch_idx)

Training step.

validation_step(batch, batch_idx, dataloader_idx)

Validation step.

Attributes#

GIMVITrainingPlan.training: bool#

Methods#

GIMVITrainingPlan.on_validation_epoch_end()[source]#

Aggregate validation step information.

GIMVITrainingPlan.training_step(batch, batch_idx)[source]#

Training step.

GIMVITrainingPlan.validation_step(batch, batch_idx, dataloader_idx)[source]#

Validation step.