`neural_compressor.experimental.data.dataloaders.pytorch_dataloader`

Initialize the Datasets class.

Module Contents

Classes

PyTorchDataLoader

PyTorchDataLoader inherits from BaseDataLoader.

class neural_compressor.experimental.data.dataloaders.pytorch_dataloader.PyTorchDataLoader(dataset, batch_size=1, last_batch='rollover', collate_fn=None, sampler=None, batch_sampler=None, num_workers=0, pin_memory=False, shuffle=False, distributed=False)[source]: PyTorchDataLoader inherits from BaseDataLoader.