模型转onnx

[Bycqg]

你好，请问一下就是怎么将weights.zip里面对应的模型文件转换成onnx格式的模型文件。

我这边转换时，使用的是transformers 官方提供的脚本convert_graph_to_onnx.py

然后转换失败了，我看了下源码文件，卡在了token = nlp.tokenizer(xxx,xxx)代码上。

这是转换的代码：python -m transformers.convert_graph_to_onnx --framework pt --model "./weights" --check-loading --quantize "./onnx/model.onnx"

[TinkingLoeng]

后来怎么实现转换？问题解决了吗

[apple2333cream]

我也是卡在了token = nlp.tokenizer(xxx,xxx)这里，请问你后来找打解决方法了吗

[wenlihaoyu]

已经支持，查到readme

[SherlockHua1995]

@Bycqg @wenlihaoyu 你好， 我也是转化不成功 python -m transformers.convert_graph_to_onnx --framework pt --model "./weights" --check-loading --quantize "./onnx/model.onnx" Error while converting the model: str object is not callable 请问可能是什么原因呢？

[cqray1990]

.

[dcdethan]

@Bycqg @wenlihaoyu 你好， 我也是转化不成功 python -m transformers.convert_graph_to_onnx --framework pt --model "./weights" --check-loading --quantize "./onnx/model.onnx" Error while converting the model: str object is not callable 请问可能是什么原因呢？

您好，请问您解决了吗

[super-tian]

python -m transformers.onnx -m E:/work/cv/trocr-chinese-main/my_train --feature=vision2seq-lm hand-write-onnx --atol 1e-4 正常转换

[HGZH990]

python -m transformers.onnx -m E:/work/cv/trocr-chinese-main/my_train --feature=vision2seq-lm hand-write-onnx --atol 1e-4 正常转换

您好，为什么我这里的--feature没有vision2seq-lm这个选项