:py:mod:`neural_compressor.experimental.data.dataloaders.pytorch_dataloader` ============================================================================ .. py:module:: neural_compressor.experimental.data.dataloaders.pytorch_dataloader .. autoapi-nested-parse:: Initialize the Datasets class. Module Contents --------------- Classes ~~~~~~~ .. autoapisummary:: neural_compressor.experimental.data.dataloaders.pytorch_dataloader.PyTorchDataLoader .. py:class:: 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) PyTorchDataLoader inherits from BaseDataLoader.