8wDlpd.png
8wDFp9.png
8wDEOx.png
8wDMfH.png
8wDKte.png

我尝试使用 hugginsface 中的 convert_graph_to_onnx.py,但出现此错误:“转换模型时出错:未安装模块 onnx!”

renji 2月前

31 0

我尝试使用此 [页面][1]python convert_graph_to_onnx.py --framework pt --model bert-base-cased bert-base-cased.onnx 中的示例使用导出模型功能,但出现错误,我需要...

我尝试使用本 [页面][1] 中的示例来使用导出模型功能

python convert_graph_to_onnx.py --framework pt --model bert-base-cased bert-base-cased.onnx

我收到一个错误,我需要使用 ONNX opset 版本 14,因此我使用了这个:

python convert_graph_to_onnx.py --framework pt --opset 14 --model bert-base-cased bert-base-cased.onnx

我收到此错误:

====== 将模型转换为 ONNX ====== convert_graph_to_onnx.py:361:FutureWarning:'transformers.convert_graph_to_onnx' 包已弃用,将在 Transformers 版本 5 中删除'
warnings.warn(ONNX opset 版本设置为:14 加载管道(模型:bert-base-cased,tokenizer:bert-base-cased)使用框架 PyTorch:2.3.1 + cpu 找到形状为{0:'batch',1:'sequence'}的输入 input_ids 找到形状为{0:'batch',1:'sequence'}的输入 token_type_ids 找到形状为{0:'batch',1:'sequence'}的输入tention_mask 找到形状为{0:'batch',1:'sequence'}的输出 output_0 找到形状为{0:'batch'}的输出 output_1 确保输入的顺序正确 position_ids 不在生成的输入列表中。生成的输入顺序:['input_ids','attention_mask','token_type_ids']转换模型时出错:未安装模块 onnx!

我已经安装了 onnx==1.16.1

有人能帮助我吗?

帖子版权声明 1、本帖标题:我尝试使用 hugginsface 中的 convert_graph_to_onnx.py,但出现此错误:“转换模型时出错:未安装模块 onnx!”
    本站网址:http://xjnalaquan.com/
2、本网站的资源部分来源于网络,如有侵权,请联系站长进行删除处理。
3、会员发帖仅代表会员个人观点,并不代表本站赞同其观点和对其真实性负责。
4、本站一律禁止以任何方式发布或转载任何违法的相关信息,访客发现请向站长举报
5、站长邮箱:yeweds@126.com 除非注明,本帖由renji在本站《tensorflow》版块原创发布, 转载请注明出处!
最新回复 (0)
  • 这个问题缺少很多信息。你引用的源页面没有链接。你克隆了 transformer 仓库并运行了脚本吗?

返回
作者最近主题: