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/versions/r2.9/api_docs/python/tf/lite/experimental/QuantizationDebugger