Data

This module contains the classes used for handling datasets and for feeding them to the Lightning trainer.

DictDataset([dictionary, feature_names, ...])	Define a torch dataset from a dictionary of lists/array/tensors and names.
DictLoader(dataset[, batch_size, shuffle])	PyTorch DataLoader for DictDataset .
DictModule(dataset[, lengths, batch_size, ...])	Lightning DataModule constructed for DictDataset .