[ONNX] onnx-graphsurgeon 이용하여 plugin 사용하기 - Group Normalization
|2020. 7. 21. 20:13
728x90
반응형
TensorRT 7.1.2 버전 부터 Group Normalization plugin 을 지원하기 시작했다.
아래 Github 에서 ONNX GraphSurgeon 을 사용할 수 있으며,
https://github.com/NVIDIA/TensorRT/tree/master/tools/onnx-graphsurgeon
NVIDIA/TensorRT
TensorRT is a C++ library for high performance inference on NVIDIA GPUs and deep learning accelerators. - NVIDIA/TensorRT
github.com
python 샘플 코드에 onnx_packnet 을 이용하여 group normalization plugin 을 추가하여 onnx 를 export 할 수 있다.
728x90
반응형
'AI Development > ONNX' 카테고리의 다른 글
| [ONNX] ONNX 배치 사이즈 변경하는 방법 + 삽질 (3) | 2021.02.02 |
|---|---|
| [ONNX] Pytorch 모델을 ONNX 모델로 변환하기 (6) | 2020.08.23 |
| [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 |