ForwardTacotron
ForwardTacotron copied to clipboard
Feature request: model compatible to export into onnx
I have made an export script in my fork but the only error that prevents me from exporting it is:
Traceback (most recent call last):
File "C:\Users\LENOVO_User\ForwardTacotron\convert-to-onnx-v2.py", line 35, in <module>
torch.onnx.export(
File "C:\Users\LENOVO_User\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\onnx\utils.py", line 506, in export
_export(
File "C:\Users\LENOVO_User\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\onnx\utils.py", line 1548, in _export
graph, params_dict, torch_out = _model_to_graph(
File "C:\Users\LENOVO_User\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\onnx\utils.py", line 1117, in _model_to_graph
graph = _optimize_graph(
File "C:\Users\LENOVO_User\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\onnx\utils.py", line 665, in _optimize_graph
graph = _C._jit_pass_onnx(graph, operator_export_type)
File "C:\Users\LENOVO_User\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\onnx\utils.py", line 1901, in _run_symbolic_function
raise errors.UnsupportedOperatorError(
torch.onnx.errors.UnsupportedOperatorError: Exporting the operator 'aten::pad_sequence' to ONNX opset version 18 is not supported. Please feel free to request support or submit a pull request on PyTorch GitHub: https://github.com/pytorch/pytorch/issues.
Is there any alternative to pad_sequence or does the model need to be adjusted to be compatible?