`neural_compressor.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.data.dataloaders.default_dataloader.default_collate(batch)[source]: Merge data with outer dimension batch size.

class neural_compressor.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)[source]: DefaultDataLoader for multiple framework backends.