OpenNMT-TF checkpoint conversion

[danaiksez]

Hi,

I would like to convert the opennmt-tf checkpoint into onnx format. I was wondering therefore if there is a way to convert either the opennmt-tf checkpoint or the saved_model.pb → huggingface model (eg .h5 format) → use huggingface optimum library .

Thank you in advance.

[ymoslem (Yasmin Moslem)]

If you do not mind the workflow, CTranslate2 is rather what you are looking for:

GitHub - OpenNMT/CTranslate2: Fast inference engine for Transformer models