neural_compressor.torch.algorithms.smooth_quant.save_load

Module Contents

Functions

recover_model_from_json(model, json_file_path, ...)

Recover ipex model from JSON file.
neural_compressor.torch.algorithms.smooth_quant.save_load.recover_model_from_json(model, json_file_path, example_inputs)[source]

Recover ipex model from JSON file.

Parameters:

  • model (object) – fp32 model need to do quantization.

  • json_file_path (json) – configuration JSON file for ipex.

  • example_inputs (tuple or torch.Tensor or dict) – example inputs that will be passed to the ipex function.

Returns:

quantized model

Return type:

(object)