728x90
반응형
Quantization Aware Training로 학습된 모델은 QuantConv2d, QuantLinear 등의 layer 가 생성되는데
이를 Standard ONNX로 변환 시켜주는 Brevitas 라이브러리가 있다.
FAQ(https://github.com/Xilinx/brevitas#faq) 를 보면 아직 알려진 이슈들은 많다.
https://github.com/Xilinx/brevitas
GitHub - Xilinx/brevitas: Brevitas: quantization-aware training in PyTorch
Brevitas: quantization-aware training in PyTorch. Contribute to Xilinx/brevitas development by creating an account on GitHub.
github.com
728x90
반응형
'AI Development > ONNX' 카테고리의 다른 글
| [ONNX] ONNX Simplifier 사용하여 모델 간소화 하기 (8) | 2021.07.26 |
|---|---|
| [ONNX] Pytorch 모델을 ONNX 모델로 변환 할 때 dynamic_axes 지정하는 방법 (0) | 2021.06.29 |
| [ONNX] ONNX 배치 사이즈 변경하는 방법 + 삽질 (3) | 2021.02.02 |
| [ONNX] Pytorch 모델을 ONNX 모델로 변환하기 (6) | 2020.08.23 |
| [ONNX] onnx-graphsurgeon 이용하여 plugin 사용하기 - Group Normalization (3) | 2020.07.21 |