словарь для указания функций отладки слоёв {function_name_str: function}, где функция принимает результат операции NumericVerify — разницу между значениями результатов операций с плавающей точкой и деквантованных операций. Функция возвращает одно скалярное значение.
model_debug_metrics
словарь для указания функций отладки модели {function_name_str: function}, где функция принимает выходные данные двух моделей и возвращает одно скалярное значение метрики (например, accuracy, IoU).
layer_direct_compare_metrics
словарь для указания функций отладки слоёв {function_name_str: function}. Сигнатура отличается от сигнатуры layer_debug_metrics: этой функции передаются (исходное значение с плавающей точкой, исходное квантованное значение, масштаб, нулевая точка). Реализация функции должна корректно деквантовать квантованное значение для сравнения. Используйте эту функцию, когда сравнения разностей недостаточно. (Примечание: квантованное значение передаётся как int8, поэтому его необходимо преобразовать в int32.)
denylisted_ops
список имён операций, которые предполагается исключить из квантования.
denylisted_nodes
список имён выходных тензоров операций, которые предполагается исключить из квантования.
fully_quantize
Логическое значение, указывающее, нужно ли полностью квантовать модель. Квантованию подвергаются не только тело модели, но также входные и выходные данные. Соответствует параметру fully_quantize функции mlir_quantize.