當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python 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

參數

  • 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

這可以運行配備調試操作的 TensorFlow Lite 轉換模型並收集調試信息。此調試器根據用戶定義的 post-processing 函數以及默認函數計算統計信息。

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.lite.experimental.QuantizationDebugger。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。