search

ThanatosShinji / onnx-tool

A parser, editor and profiler tool for ONNX models.
https://pypi.org/project/onnx-tool/
MIT License
383 stars 51 forks source link

Need support for LayerNormalization #41

Open haojin2 opened 1 year ago

haojin2 commented 1 year ago

I've got a model that has LayerNormalization layer in it, and when I'm running profile the tool gives the following warning

UserWarning: node LayerNormalization is not registed for profiling, return 0 Macs and 0 params as default. Use NODEPROFILER_REGISTRY to register your profiler for this node.
ThanatosShinji commented 1 year ago

code is done, but I can't find a model to validate it. maybe you can checkout the newest commit to profile your model.