이것저것 공부한 기록

[Tensorflow] BackendIsNotSupposedToImplementIt: LayerNormalization is not implemented 본문

Others

[Tensorflow] BackendIsNotSupposedToImplementIt: LayerNormalization is not implemented

블랜디 2024. 2. 5. 10:22

요즘 Pytorch model을 tensorflow lite로 변환하기 위해 우당탕탕 중인데..

 

onnx_tf.backend의 export_graph함수 호출 시 위와 같은 에러가 발생했다.

 

https://github.com/onnx/onnx-tensorflow/issues/1059

 

BackendIsNotSupposedToImplementIt: LayerNormalization is not implemented · Issue #1059 · onnx/onnx-tensorflow

Describe the bug I face this issue when converting onnx model to tensorflow model. Please help. Thank you for your all recommendations. To Reproduce tf_model_path = "tf_model" tf_rep.export_graph(t...

github.com

 

+)해당 discussion에는 onnx파일 생성 전 onnx버전을 1.13.1로 맞추라고만 되어 있는데,

onnx2tf.convert 실행 전에 onnx 버전을 1.15.0으로 다시 업그레이드 해주어야 함 (1.13.1 버전에는 없는 class를 사용함)

 

 

++)

colab에서 할 때 onnx_tf에서 에러가 나서 위 library를 사용하려고 했으나 알수 없는 이유로 input tensor array가 변경되었음.

(1, 80, 3000이 1, 3000, 80이 되어버리는..)

그냥 로컬 gpu 환경에서 돌렸더니 onnx_tf 에러가 해결되어 변환 완료함