I am trying to export ONNX model using torch.onnx.export(torch.jit.script(model), dynamo=False)
How can I figure out which tuples it can not lower. Model is very complex and most of the forward method returns tuples.
Error:
File "/home/himalay.joriwal/.conda/envs/uat/lib/python3.11/site-packages/torch/onnx/__init__.py", line 375, in export
export(
File "/home/himalay.joriwal/.conda/envs/uat/lib/python3.11/site-packages/torch/onnx/utils.py", line 502, in export
_export(
File "/home/himalay.joriwal/.conda/envs/uat/lib/python3.11/site-packages/torch/onnx/utils.py", line 1564, in _export
graph, params_dict, torch_out = _model_to_graph(
^^^^^^^^^^^^^^^^
File "/home/himalay.joriwal/.conda/envs/uat/lib/python3.11/site-packages/torch/onnx/utils.py", line 1117, in _model_to_graph
graph = _optimize_graph(
^^^^^^^^^^^^^^^^
File "/home/himalay.joriwal/.conda/envs/uat/lib/python3.11/site-packages/torch/onnx/utils.py", line 581, in _optimize_graph
_C._jit_pass_lower_all_tuples(graph)
RuntimeError: Couldn't lower all tuples