XQuant Report Pytorch¶

model_compression_toolkit.xquant.pytorch.facade_xquant_report.xquant_report_pytorch_experimental(float_model, quantized_model, repr_dataset, validation_dataset, xquant_config)¶

Generate an explainable quantization report for a quantized Pytorch model.

Parameters:

• float_model (torch.nn.Module) – The original floating-point Pytorch model.

• quantized_model (torch.nn.Module) – The quantized Pytorch model.

• repr_dataset (Callable) – The representative dataset used during quantization.

• validation_dataset (Callable) – The validation dataset used for evaluation.

• xquant_config (XQuantConfig) – Configuration settings for explainable quantization.

Returns:

A dictionary containing the collected similarity metrics and report data.

Return type:

Dict[str, Any]