Optimum 文件
TFLite 匯出配置類
並獲得增強的文件體驗
開始使用
TFLite 匯出配置類
基礎類
類 optimum.exporters.tflite.TFLiteConfig
< 來源 >( 配置: PretrainedConfig 任務: str 批次大小: int = 1 序列長度: typing.Optional[int] = None 選擇數量: typing.Optional[int] = None 寬度: typing.Optional[int] = None 高度: typing.Optional[int] = None 通道數量: typing.Optional[int] = None 特徵大小: typing.Optional[int] = None 最大幀數: typing.Optional[int] = None 音訊序列長度: typing.Optional[int] = None 點批次大小: typing.Optional[int] = None 每影像點數: typing.Optional[int] = None 視覺序列長度: typing.Optional[int] = None )
TFLite 可匯出模型的基礎類，描述瞭如何透過 TFLite 格式匯出模型的元資料。
類屬性
NORMALIZED_CONFIG_CLASS (
Type) — 派生自 NormalizedConfig 的類，指定如何標準化模型配置。DUMMY_INPUT_GENERATOR_CLASSES (
Tuple[Type]) — 派生自 DummyInputGenerator 的類元組，指定如何建立虛擬輸入。ATOL_FOR_VALIDATION (
Union[float, Dict[str, float]]) — 一個浮點數或將任務名稱對映到浮點數的字典，其中浮點值表示模型轉換驗證期間使用的絕對容差值。MANDATORY_AXES (
Tuple[Union[str, Tuple[Union[str, Tuple[str]]]]]) — 一個元組，其中每個元素要麼是- 軸名稱，例如“batch_size”或“sequence_length”，表示匯出模型需要該軸維度，
- 或者包含兩個元素的元組
- 第一個是字串或字串元組，指定需要該軸的任務
- 第二個是軸名稱。
例如：
MANDATORY_AXES = ("batch_size", "sequence_length", ("multiple-choice", "num_choices"))表示匯出模型時始終需要指定批次大小和序列長度值，並且當任務是多項選擇時需要指定可能選擇的數量。
包含提供給模型的輸入張量軸定義的字典。
包含提供給模型的輸出張量軸定義的字典。
中端類
類 optimum.exporters.tflite.config.TextEncoderTFliteConfig
< 來源 >( 配置: PretrainedConfig 任務: str 批次大小: int = 1 序列長度: typing.Optional[int] = None 選擇數量: typing.Optional[int] = None 寬度: typing.Optional[int] = None 高度: typing.Optional[int] = None 通道數量: typing.Optional[int] = None 特徵大小: typing.Optional[int] = None 最大幀數: typing.Optional[int] = None 音訊序列長度: typing.Optional[int] = None 點批次大小: typing.Optional[int] = None 每影像點數: typing.Optional[int] = None 視覺序列長度: typing.Optional[int] = None )
處理基於編碼器的文字架構。
類 optimum.exporters.tflite.config.VisionTFLiteConfig
< 來源 >( 配置: PretrainedConfig 任務: str 批次大小: int = 1 序列長度: typing.Optional[int] = None 選擇數量: typing.Optional[int] = None 寬度: typing.Optional[int] = None 高度: typing.Optional[int] = None 通道數量: typing.Optional[int] = None 特徵大小: typing.Optional[int] = None 最大幀數: typing.Optional[int] = None 音訊序列長度: typing.Optional[int] = None 點批次大小: typing.Optional[int] = None 每影像點數: typing.Optional[int] = None 視覺序列長度: typing.Optional[int] = None )
處理視覺架構。