Datamodule

visionlab.CifarDataModule

Bases: LightningDataModule

A custom LightningDataModule

prepare_data()

prepares data for the dataloaders

setup(stage=None)

used by trainer to setup the dataset for training and evaluation

test_dataloader()

the dataloader used during testing

train_dataloader()

the dataloader used during training

val_dataloader()

the dataloader used during validation