neural_compressor.ux.components.optimization.pruning.optimize_model
¶
Pruning optimization script.
Module Contents¶
Functions¶
|
Parse input arguments. |
|
Execute pruning optimization. |
- neural_compressor.ux.components.optimization.pruning.optimize_model.parse_args() Any ¶
Parse input arguments.
- neural_compressor.ux.components.optimization.pruning.optimize_model.optimize_model(input_graph: str, output_graph: str, framework: str, config: str) None ¶
Execute pruning optimization.