Udpipe 2 Converting Models to Onnx

ufal / udpipe

UDPipe: Trainable pipeline for tokenizing, tagging, lemmatizing and parsing Universal Treebanks and other CoNLL-U files

Mozilla Public License 2.0

355 stars 74 forks source link

Udpipe 2 Converting Models to Onnx #142

Open RobertGengiu opened 3 years ago

RobertGengiu commented 3 years ago

Hi, Has anyone tried to convert a model? If yes, what are the steps necessary to do it?

Thank you!

foxik commented 3 years ago

I do not know about anyone who tried. Note that we only store the weights in the released checkpoints, the graphs are generated by the source itself -- so maybe one should try to convert the fully-loaded models, or to save them including the computation graph first? But I have no experience with the Onnx myself.