XQuant Configuration¶

class model_compression_toolkit.xquant.common.xquant_config.XQuantConfig(report_dir, custom_similarity_metrics=None)¶

Configuration for generating the report. It allows to set the log dir that the report will be saved in and to add similarity metrics to measure between tensors of the two models.

Initializes the configuration for explainable quantization.

Parameters:

• report_dir (str) – Directory where the reports will be saved.

• custom_similarity_metrics (Dict[str, Callable]) – Custom similarity metrics to be computed between tensors of the two models. The dictionary keys are similarity metric names and the values are callables that implement the similarity metric computation.