跳到正文
Hugging Face Blog·· 2022-06-22AI 评分28

如何用 Hugging Face Optimum 将 Transformers 模型转换为 ONNX

Convert Transformers to ONNX with Hugging Face Optimum

AI 导读

Hugging Face 博客介绍了三种将 Transformers 模型转为 ONNX 的方式,从底层 torch.onnx、中层 transformers.onnx 到高层 Optimum,均以 distilbert-base-uncased-finetuned-sst-2-english 文本分类模型为例。

来源:Hugging Face Blog · huggingface.co