burn
burn copied to clipboard
[Example] Add custom images dataset
Currently we only have an example using the MNISTDataset via the huggingface loader.
Would be nice to include an example to load a custom image dataset from disk, similar to torchvision.datasets.DatasetFolder for image classification.