search

google-ai-edge / model-explorer

A modern model graph visualizer and debugger
Apache License 2.0
991 stars 74 forks source link

Large challenging graph to display #50

Open justinchuby opened 4 months ago

justinchuby commented 4 months ago

The UI keeps displaying "Processing please wait" when trying to expand the graph. The graph has ~32k nodes.

image

Steps

pip install model_explorer_onnx==0.1.4
model-explorer model_1.onnx --extensions=model_explorer_onnx

Model: https://drive.google.com/file/d/1R3vW92eSLbtolH53HBlhG2CdEQPs_PO6/view obtained from https://github.com/microsoft/onnxruntime/issues/20354

jinjingforever commented 4 months ago

Thank you for trying this out!

Your observation is correct. For super large graphs like the one you linked, the "processing please wait" message will often appear when you expand or collapse a layer because Model Explorer needs to layout all the layers affected by it, especially when there are multiple expanded layers.

We will explore some more ways to make this process faster. (And thanks for providing a great model for us to test!)