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 | Необязательно, использовать преобразователь вместо квантованной модели. |
| Исключения | |
|---|---|
ValueError | Если отладчик не удалось создать. |
| Атрибуты | |
|---|---|
options | |
Методы
get_debug_quantized_model
get_debug_quantized_model() -> bytes
Возвращает инструментированную квантованную модель.
Преобразует квантованную модель с инициализированным конвертером и возвращает байты модели. Модель будет инструментирована операциями числовой проверки и должна использоваться только для отладки.
| Возвращает | |
|---|---|
| Байты модели, соответствующие модели. |
| Исключения | |
|---|---|
ValueError | если конвертер не передан в отладчик. |
get_nondebug_quantized_model
get_nondebug_quantized_model() -> bytes
Возвращает неинструментированную квантованную модель.
Преобразует квантованную модель с инициализированным конвертером и возвращает байты модели без отладки. Модель не будет инструментирована операциями числовой проверки.
| Возвращает | |
|---|---|
| Байты модели, соответствующие модели. |
| Исключения | |
|---|---|
ValueError | если конвертер не передан в отладчик. |
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