neural_compressor.ux.utils.workload.tuning

Configuration tuning module.

Module Contents

Classes

Strategy	Configuration Strategy class.
MultiObjectives	Configuration MultiObjectives class.
AccCriterion	Configuration AccCriterion class.
ExitPolicy	Configuration ExitPolicy class.
Workspace	Configuration Workspace class.
Diagnosis	Configuration Diagnosis class.
Tuning	Configuration Tuning class.

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

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration Strategy class.

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

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration MultiObjectives class.

property objective: List[str]

Get objectives.

property weight: List[float]

Get weights.

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

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration AccCriterion class.

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

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration ExitPolicy class.

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

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration Workspace class.

class neural_compressor.ux.utils.workload.tuning.Diagnosis(data: Dict[str, Any] | None = None)

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration Diagnosis class.

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

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration Tuning class.

set_timeout(timeout: int) → None

Update tuning timeout in config.

set_max_trials(max_trials: int) → None

Update max tuning trials in config.

set_performance_only(performance_only: Any) → None

Update performance only flag in config.

set_random_seed(random_seed: int) → None

Update random seed value in config.

set_workspace(path: str) → None

Update tuning workspace path in config.