pytorch-crf model to onnx conversion

[9lan9]

Actually, it is the same issue in https://github.com/pytorch/pytorch/issues/39245, any solution for this one?

[kmkurn]

I personally never used ONNX, so I don't think I can help, sorry! I'll keep the issue open in case anyone else can help.

[9lan9]

I personally never used ONNX, so I don't think I can help, sorry! I'll keep the issue open in case anyone else can help.

ok, thanks.

[sralvins]

output of _viterbi_decode is a list of int lists with variable lengths which is invalid for onnx output. just convert it to a tensor. for example:

[9lan9]

output of _viterbi_decode is a list of int lists with variable lengths which is invalid for onnx output. just convert it to a tensor. for example:

convert list to tensor would cause problems, that's the problem

[SeanLiaoy]

Cannot convert the CRF layer to ONNX even using the latest version of PyTorch. I guess the combination of some operators may cause issues in PyTorch converter. And it also cannot be converted to torchscript. Here is the version of CRF that can be converted torchscript: https://github.com/facebookresearch/pytext/blob/master/pytext/models/crf.py . Unfortunately, CRF code provided by pytext also cannot be converted to ONNX

[etern]

pytorch model is converted to ONNX via torchscript, this issue is related #79

[Ulitochka]

https://github.com/taishan1994/pytorch_bert_bilstm_crf_ner/tree/16ee44df462dfa9fde13597d724ea9f6477ef935/convert_onnx