Open ZJaume opened 3 years ago
Sorry, forgot the versions.
keras2onnx==1.7.0
onnx==1.9.0
onnxconverter-common==1.8.1
onnxruntime==1.7.0
tensorflow==2.3.2
transformers==4.4.2
Try this converter, it has worked for me: https://github.com/onnx/onnxmltools . Although you should take into account that I used tensorflow 2.2 for training the model, so make this adjustment also.
Yeah, I tried tf2onnx (which seems to be wrapped in that tool) and it worked.
Are text classifiaction huggingface models supported? I'm getting invalid graph when running this: