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 )
处理视觉架构。