neural_compressor.ux.components.optimization.tune.tuning

Tuning class.

Module Contents

Classes

Tuning

Tuning class.

AccuracyCriterion

Interface for accuracy criterion.

TuningDetails

Interface for tuning details.

class neural_compressor.ux.components.optimization.tune.tuning.Tuning(optimization_data: dict, project_data: dict, dataset_data: dict)

Bases: neural_compressor.ux.components.optimization.optimization.Optimization

Tuning class.

property optimization_script: str

Get optimization script path.

property configuration_data: dict

Get configuration data for tuning config generator.

property parameters: List[str]

Get optimization parameters.

execute() None

Execute tuning.

generate_config() None

Generate yaml config.

class neural_compressor.ux.components.optimization.tune.tuning.AccuracyCriterion

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Interface for accuracy criterion.

class neural_compressor.ux.components.optimization.tune.tuning.TuningDetails(data: dict | None = None)

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Interface for tuning details.