[MLOps] Transformers로 학습한 모델 ONNX로 변환하기

ONNX란?

ONNX는 Open Neural Network eXchange의 약자로 PyTorch, TensorFlow 등 다양한 딥러닝 프레임 워크에서 학습된 모델을 상호 호환 가능하도록 해주는 일종의 deep learning model 포멧이다.
즉, PyTorch에서 학습한 모델을 ONNX로 export한 모델을 TensorFlow에서 import하여 사용할 수 있으며 그 반대도 가능하다.

지원하는 아키텍쳐 export 하기

지원하는 모델 확인하기

지원하지 않는 아키텍쳐 export 하기

1. custom ONNX configuration 작성하기
2. 모델을 ONNX로 export 하기

custom ONNX configuration 작성하기