Conversion to ONNX

[TeddyAlbina]

Hello

I'm trying to convert this model to ONNX with no success, using the ExpansionNet_v2 script convert2onnx.py.

python convert2onnx.py --load_model_path mod/kaz_model.pth --output_onnx_path ./kaz_model.onnx

I got this error when trying to do so, do anyone have an idea on how to fix it

RuntimeError: The size of tensor a (10000) must match the size of tensor b (18365) at non-singleton dimension 0

Thank you.

[shahizat]

hi @TeddyAlbina, please change max sequence length to 63. Default is 74. Hope it helps.