On-device에서 딥러닝 모델을 이용하기 위해 pytorch 모델을 tensorflow lite로 변환하는 방법을 정리해 보겠습니다. 주로 딥러닝 학습은 python에서 pytorch를 이용하여 수행됩니다. 하지만 모바일 또는 엣지 디바이스에서 이용하려면 C++이나 JAVA에서 딥러닝 모델을 불러와야 합니다. Tensorflow lite를 통해 C++, JAVA, Kotlin, Swift에서 딥러닝 모델을 사용할 수 있어 온디바이스를 개발에 많이 이용됩니다. 1. torch-> onnxonnx는 torch, tensorflow와 같은 서로 다른 딥러닝 프레임워크를 호환되게 사용할 수 있도록 만들어진 공유입니다. 우선 torch model을 onnx로 저장합니다. 이때 모델 그래프와 가중치가 함께 저..