tf.lite.experimental.QuantizationDebugger
Отладчик моделей TensorFlow Lite в режиме отладки с квантованием.
tf.lite.experimental.QuantizationDebugger(
quant_debug_model_path: Optional[str] = None,
quant_debug_model_content: Optional[bytes] = None,
float_model_path: Optional[str] = None,
float_model_content: Optional[bytes] = None,
debug_dataset: Optional[Callable[[], Iterable[Sequence[np.ndarray]]]] = None,
debug_options: Optional[tf.lite.experimental.QuantizationDebugOptions] = None,
converter: Optional[TFLiteConverter] = None
) -> None
Используется в блокнотах
| Используется в учебных материалах |
|---|
Этот инструмент запускает преобразованные модели TensorFlow Lite, оснащённые операциями отладки, и собирает отладочную информацию. Отладчик вычисляет статистику с помощью пользовательских функций постобработки, а также функций по умолчанию.
| Аргументы | |
|---|---|
quant_debug_model_path | Путь к файлу отладочной модели TFLite с квантованием. |
quant_debug_model_content | Содержимое отладочной модели TFLite с квантованием. |
float_model_path | Путь к файлу модели TFLite с вещественными числами. |
float_model_content | Содержимое модели TFLite с вещественными числами. |
debug_dataset | фабричная функция, возвращающая генератор набора данных, который используется для создания входных примеров (списка np.ndarray) для модели. Типы и форма созданных элементов должны совпадать с типами и формой входных данных модели. |
debug_options | Параметры отладки для указанной модели. |
converter | Необязательный параметр: использовать converter вместо модели с квантованием. |
| Исключения | |
|---|---|
ValueError | Если не удалось создать отладчик. |
| Атрибуты | |
|---|---|
options | |
Методы
get_debug_quantized_model
get_debug_quantized_model() -> bytes
Возвращает инструментированную модель с квантованием.
Преобразует модель с квантованием с помощью инициализированного converter и возвращает байты модели. Модель будет инструментирована операциями числовой проверки и должна использоваться только для отладки.
| Возвращает | |
|---|---|
| Байты, соответствующие модели. |
| Исключения | |
|---|---|
ValueError | если отладчику не передан converter. |
get_nondebug_quantized_model
get_nondebug_quantized_model() -> bytes
Возвращает неинструментированную модель с квантованием.
Преобразует модель с квантованием с помощью инициализированного converter и возвращает байты модели без отладки. Модель не будет инструментирована операциями числовой проверки.
| Возвращает | |
|---|---|
| Байты, соответствующие модели. |
| Исключения | |
|---|---|
ValueError | если отладчику не передан converter. |
layer_statistics_dump
layer_statistics_dump(
file: IO[str]
) -> None
Записывает статистику слоёв в файл в формате CSV.
| Аргументы | |
|---|---|
file | файл или файловый объект для записи. |
run
run() -> None
Запускает модели и получает метрики.
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/lite/experimental/QuantizationDebugger