SciomageLAB 2024. 10. 18. 00:15
반응형

brevitas는 라틴어로 '짧음' 이라는 뜻이다. 어떤 느낌으로 양자화된 딥러닝 모델 오픈소스 이름을 지었는지 알 것 같다.

Brevitas는 양자화-인식 훈련(QAT, quantization-aware training)을 위한 PyTorch 연구 라이브러리입니다. Torch 연구 라이브러리 인데, 오픈소스는 xilinx에 있습니다.

공식 Xilinx 제품이 아니라 연구 프로젝트입니다. 2021년 1월 30일부터 시작해서, 지금(작성일 2021-03-31 기준) 최신 프로젝트입니다.

https://github.com/Xilinx/brevitas/

PyTorch와 같은 딥러닝 프레임워크인데, onnx 표준으로 export 할 수도 있고, Xilinx FINN으로도 export 할 수 있다. Torch도 양자화 기능이 있긴 한데 CPU만 지원한다. FPGA를 지원하지 않기 때문에 onnx 표준이나 FINN으로 export 할 수 있도록 연구 하는거 같다.

반응형