1.7 KiB
1.7 KiB
导出 🤗 Transformers 模型到 ONNX
🤗 Transformers提供了一个transformers.onnx包，通过利用配置对象，您可以将模型checkpoints转换为ONNX图。
有关更多详细信息，请参阅导出 🤗 Transformers 模型的指南。
ONNX Configurations
我们提供了三个抽象类，取决于您希望导出的模型架构类型：
- 基于编码器的模型继承 [
~onnx.config.OnnxConfig] - 基于解码器的模型继承 [
~onnx.config.OnnxConfigWithPast] - 编码器-解码器模型继承 [
~onnx.config.OnnxSeq2SeqConfigWithPast]
OnnxConfig
autodoc onnx.config.OnnxConfig
OnnxConfigWithPast
autodoc onnx.config.OnnxConfigWithPast
OnnxSeq2SeqConfigWithPast
autodoc onnx.config.OnnxSeq2SeqConfigWithPast
ONNX Features
每个ONNX配置与一组 特性 相关联，使您能够为不同类型的拓扑结构或任务导出模型。
FeaturesManager
autodoc onnx.features.FeaturesManager