我尝试使用此 [页面][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
有人能帮助我吗?