Export trained model to ONNX

[RahulJha11]

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]

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