neural_compressor.ux.utils.workload.dataloader

Configuration dataloader module.

Module Contents

Classes

Dataset	Configuration Dataset class.
LabelBalance	Configuration LabelBalance class.
Filter	Configuration Filter class.
Transform	Configuration Transform class.
Dataloader	Configuration Dataloader class.

class neural_compressor.ux.utils.workload.dataloader.Dataset(name: str, data: Dict[str, Any] = {})

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration Dataset class.

serialize(serialization_type: str = 'default') → Dict[str, Any]

Serialize Dataset class.

is_dummy() → bool

Check if dummy dataloader is used.

class neural_compressor.ux.utils.workload.dataloader.LabelBalance(data: Dict[str, Any] = {})

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration LabelBalance class.

class neural_compressor.ux.utils.workload.dataloader.Filter(data: Dict[str, Any] = {})

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration Filter class.

class neural_compressor.ux.utils.workload.dataloader.Transform(name: str, parameters: Dict[str, Any] = {})

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration Transform class.

serialize(serialization_type: str = 'default') → Dict[str, Any]

Serialize Transform class.

class neural_compressor.ux.utils.workload.dataloader.Dataloader(data: Dict[str, Any] = {})

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration Dataloader class.

set_transforms_from_list(transforms_list: List[dict]) → None

Set transformations ordered dict from list.

set_dataset(dataset_data: Dict[str, Any]) → None

Set dataset for dataloader.

serialize(serialization_type: str = 'default') → Dict[str, Any]

Serialize Dataloader class.