Internal error when reording ONNX operators

[leiwen83]

Hi,

I try to take some test over some onnx model, but meet the error as "internal error when reording onnx operators"

Traceback (most recent call last):
File "convert.py", line 4, in
old_model = taso.load_onnx("part4_wo_stack.onnx")
File "/opt/conda/lib/python3.7/site-packages/taso-0.1.0-py3.7-linux-x86_64.egg/taso/init.py", line 728, in load_onnx
assert len(node_list) == len(model.graph.node), "Internal error when reording ONNX operators"
AssertionError: Internal error when reording ONNX operators

The method I use is like the README.md says, and the environment I use is prebuit tasoml/cuda100 docker:

import taso
import onnx

old_model = taso.load_onnx("/path/to/load/onnx/model")
taso_graph = taso.optimize(old_model)
new_model = taso.export_onnx(taso_graph)
onnx.save(new_model, "/path/to/save/new/onnx/model")

Is there some clue for how to debug this?

Thx,
Lei

[Sagebati]

Hi @leiwen83, did you find any solution ? I have this issue as well
With https://github.com/onnx/models/blob/master/vision/classification/alexnet/model/bvlcalexnet-9.onnx
using the docker environment too.