neural_compressor.experimental.data.dataloaders.default_dataloader

Default dataloader for multiple framework backends.

Module Contents

Classes

DefaultDataLoader

DefaultDataLoader for multiple framework backends.

Functions

default_collate(batch)

Merge data with outer dimension batch size.

neural_compressor.experimental.data.dataloaders.default_dataloader.default_collate(batch)

Merge data with outer dimension batch size.

class neural_compressor.experimental.data.dataloaders.default_dataloader.DefaultDataLoader(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.experimental.data.dataloaders.base_dataloader.BaseDataLoader

DefaultDataLoader for multiple framework backends.

property dataloader

Return dataloader.

batch(batch_size, last_batch='rollover')

Set batch_size and last_batch.