neural_compressor.data.dataloaders.default_dataloader

Default dataloader for multiple framework backends.

Classes

DefaultDataLoader

DefaultDataLoader for multiple framework backends.

Functions

default_collate(batch)

Merge data with outer dimension batch size.

Module Contents

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

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)

DefaultDataLoader for multiple framework backends.