neural_compressor.data.dataloaders.pytorch_dataloader

Initialize the DATASETS class.

Classes

PyTorchDataLoader

PyTorchDataLoader inherits from BaseDataLoader.

Module Contents

class neural_compressor.data.dataloaders.pytorch_dataloader.PyTorchDataLoader[source]: PyTorchDataLoader inherits from BaseDataLoader.