[ONNX] onnx-graphsurgeon 이용하여 plugin 사용하기 - Group Normalization

728x90

반응형

 

TensorRT 7.1.2 버전 부터 Group Normalization plugin 을 지원하기 시작했다.

 

 

 

아래 Github 에서 ONNX GraphSurgeon 을 사용할 수 있으며, 

 

https://github.com/NVIDIA/TensorRT/tree/master/tools/onnx-graphsurgeon

NVIDIA/TensorRT

 

 

 

python 샘플 코드에 onnx_packnet 을 이용하여 group normalization plugin 을 추가하여 onnx 를 export 할 수 있다. 

 

 

 

 

 

 

 

 

728x90

반응형

저작자표시비영리

• 카카오스토리
• 트위터
• 페이스북

'AI Development > ONNX' 카테고리의 다른 글

[ONNX] ONNX 배치 사이즈 변경하는 방법 + 삽질  (3)	2021.02.02
[ONNX] Pytorch 모델을 ONNX 모델로 변환하기  (4)	2020.08.23
[ONNX] onnx-graphsurgeon 이용하여 plugin 사용하기 - Group Normalization  (3)	2020.07.21
[ONNX] Netron : ONNX model Visualization  (0)	2020.07.16
[Onnx] Onnx Tutorials (정리중)  (0)	2020.05.14
[ONNX] Pytorch 에서 Onnx 로 변환 시 Gather op 때문에 export 안되는 문제  (5)	2020.04.28