search

ZhangGe6 / onnx-modifier

A tool to modify ONNX models in a visualization fashion, based on Netron and Flask.
MIT License
1.28k stars 157 forks source link

Will the operator support layernorm and other transformation operators such as permute and roll? #86

Closed wangxdgg closed 9 months ago

wangxdgg commented 10 months ago

Hi Zhang, Will the operator support layernorm and other transformation operators such as permute and roll? thank u:)

ZhangGe6 commented 10 months ago

@wangxdgg Hi, sorry for the late update and response. I have uploaded a new version which you can find layerNormalization operator in the Adding node selector. Please ensure to export the ONNX model (from torch or other frameworks) with opset >=17, in which layerNormalization is added into ONNX operators.

As the official ONNX operators doc states, the permute operation can be realized using DepthToSpace operator. The roll operation may need more investigation.

ZhangGe6 commented 9 months ago

Closing since no activity for more than 3 weeks. Feel free for more discussions if any problem exists.