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

Ellie·2023년 2월 19일
0

MLOps

목록 보기
3/3

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 작성하기

profile
A bit nerdy

0개의 댓글