neural_compressor.ux.components.optimization.execute_optimization

Execute tune.

Module Contents

Functions

execute_optimization_pytorch_script(→ Any)	Optimize PyTorch scripted models.
execute_optimization_regular(→ Any)	Optimize PB/ONNX models.
execute_optimization(→ dict)	Get configuration.
parse_model_data_to_bench_names(→ dict)	Parse names to Bench format.
parse_logs(→ dict)	Parse optimization logs.
check_if_collect_tuning_history(→ bool)	Check whether tuning history can be collected for specified optimization.

neural_compressor.ux.components.optimization.execute_optimization.execute_optimization_pytorch_script(optimization: Any, request_id: Any, optimization_id: Any) → Any

Optimize PyTorch scripted models.

neural_compressor.ux.components.optimization.execute_optimization.execute_optimization_regular(optimization: Any, request_id: Any, optimization_id: Any) → Any

Optimize PB/ONNX models.

neural_compressor.ux.components.optimization.execute_optimization.execute_optimization(data: Dict[str, Any]) → dict

Get configuration.

neural_compressor.ux.components.optimization.execute_optimization.parse_model_data_to_bench_names(data: dict) → dict

Parse names to Bench format.

neural_compressor.ux.components.optimization.execute_optimization.parse_logs(optimization: neural_compressor.ux.components.optimization.optimization.Optimization, optimization_details: dict, project_details: dict, logs: List[str]) → dict

Parse optimization logs.

neural_compressor.ux.components.optimization.execute_optimization.check_if_collect_tuning_history(optimization: neural_compressor.ux.components.optimization.optimization.Optimization) → bool

Check whether tuning history can be collected for specified optimization.