You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.
What issue meet, what's expected?:
Because NNI does not have special treatment for TupleUnpack and ListUnpack operations, TorchModuleGraph will have a graph construction error in some models with TupleUnpack and ListUnpack operations(such as retina face/shufflenet). I want to support these operations in the next release and fix this problem.
How to reproduce it?:
Take shufflenet as an example. The TorchModuleGraph will build the graph in the wrong way(as shown in the following img).
In contrast, the right way to build the graph of the shufflenet is shown in the following graph.
Additional information:
The text was updated successfully, but these errors were encountered:
Environment:
What issue meet, what's expected?:
Because NNI does not have special treatment for TupleUnpack and ListUnpack operations, TorchModuleGraph will have a graph construction error in some models with TupleUnpack and ListUnpack operations(such as retina face/shufflenet). I want to support these operations in the next release and fix this problem.
How to reproduce it?:
Take shufflenet as an example. The TorchModuleGraph will build the graph in the wrong way(as shown in the following img).
In contrast, the right way to build the graph of the shufflenet is shown in the following graph.
Additional information:
The text was updated successfully, but these errors were encountered: