Export trained model to ONNX

wenwenyu / PICK-pytorch

Code for the paper "PICK: Processing Key Information Extraction from Documents using Improved Graph Learning-Convolutional Networks" (ICPR 2020)

https://arxiv.org/abs/2004.07464

MIT License

553 stars 191 forks source link

Export trained model to ONNX #124

Open RahulJha11 opened 1 year ago

RahulJha11 commented 1 year ago

How to export the Pick-Pytorch model into ONNX. I try but get an error because the keyword argument is used in forward(). If anyone converted the model into ONNX, please give some suggestions it will very helpful to me. Thank!

xuanhaihust commented 1 year ago

you should divide the original model into some parts, for example: encoder, decoder... And then export each part to ONNX.