Vision test

VisionDataModuleTests

Bases: DataModuleTests[VisionDataModuleType]

Tests for a datamodule/dataset for vision tasks.

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