neural_compressor.ux.components.optimization.tune.tune_model

Generic tuning script.

Module Contents

Functions

parse_args(→ Any)	Parse input arguments.
tune_model(→ None)	Execute tuning.
set_eager_execution(→ None)	Set eager execution as required by model.

neural_compressor.ux.components.optimization.tune.tune_model.parse_args() → Any

Parse input arguments.

neural_compressor.ux.components.optimization.tune.tune_model.tune_model(input_graph: str, output_graph: str, config: str, framework: str) → None

Execute tuning.

neural_compressor.ux.components.optimization.tune.tune_model.set_eager_execution(input_graph: str) → None

Set eager execution as required by model.