[ONNX] Brevitas, QAT 모델을 Standard ONNX 모델로 생성하는 라이브러리

 

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