Skip to content

Image classification test

ImageClassificationDataModuleTests #

Bases: VisionDataModuleTests[ImageClassificationDataModuleType]

Tests for a datamodule/dataset for image classification.

This is a simple data regression test for now. For each of the train_dataloader, valid_dataloader