Core
lab.trainer.LabTrainer
Bases: Trainer
lab.module.LabModule
Bases: LightningModule
a custom PyTorch Lightning LightningModule