search

ZhangGe6 / onnx-modifier

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

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

Closed wangxdgg closed 11 months ago

wangxdgg commented 1 year ago

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

ZhangGe6 commented 1 year 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 11 months ago

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