Optimum 文档

TFLite 导出配置类

您正在查看的是需要从源码安装。如果您想进行常规 pip 安装,请查看最新的稳定版本 (v1.27.0)。
Hugging Face's logo
加入 Hugging Face 社区

并获得增强的文档体验

开始使用

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 )

参数

  • config (transformers.PretrainedConfig) — 模型配置。
  • task (str, 默认为 "feature-extraction") — 模型应导出的任务。
  • 其余参数用于指定模型可以接受的输入形状。 —
  • 是否需要它们取决于 TFLiteConfig 为其设计的模型。 —

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")) 表示导出模型时始终需要指定批量大小和序列长度值,并且当任务是多项选择时需要指定可能选择的数量。

输入

< >

( ) Dict[str, Dict[int, str]]

返回

Dict[str, Dict[int, str]]

每个输入名称到轴位置映射的映射到轴符号名称。

包含提供给模型的输入张量轴定义的字典。

输出

< >

( ) Dict[str, Dict[int, str]]

返回

Dict[str, Dict[int, str]]

每个输出名称到轴位置映射的映射到轴符号名称。

包含提供给模型的输出张量轴定义的字典。

生成虚拟输入

< >

( )

中端类

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 )

处理视觉架构。

< > 在 GitHub 上更新