`neural_compressor.data.dataloaders.mxnet_dataloader`¶

MXNet Dataloader implementation.

Module Contents¶

Classes¶

MXNetDataLoader

Subclass of BaseDataLoader.

class neural_compressor.data.dataloaders.mxnet_dataloader.MXNetDataLoader(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)¶

Bases: neural_compressor.data.dataloaders.base_dataloader.BaseDataLoader

Subclass of BaseDataLoader.

neural_compressor.data.dataloaders.mxnet_dataloader¶

Module Contents¶

Classes¶

`neural_compressor.data.dataloaders.mxnet_dataloader`¶