반응형

TensorFlow 3

PyTorch 모델을 ONNX로 변환하고 ONNX Runtime으로 배포하는 방법

딥러닝 모델을 연구하고 개발할 때 PyTorch는 강력하고 유연한 환경을 제공합니다. 하지만 실제 서비스 환경에 모델을 배포하거나, 모바일 및 임베디드 기기처럼 자원이 제한된 환경에서 모델을 구동할 때는 프레임워크의 무거운 무게가 걸림돌이 되기도 합니다. 이러한 이종 환경 간의 장벽을 허물고 배포 효율성을 극대화하기 위해 개발된 표준이 바로 ONNX입니다. 이번 글에서는 PyTorch 모델을 ONNX 포맷으로 변환하고, ONNX Runtime을 통해 고성능 인퍼런스를 구현하는 방법을 구체적으로 알아보겠습니다.핵심 요약 3줄ONNX는 서로 다른 딥러닝 프레임워크 간에 모델을 공유하고 실행할 수 있도록 지원하는 개방형 표준 포맷입니다.PyTorch의 torch.onnx.export 함수를 사용하면 몇 줄의 ..

모델 저장 및 로드 방법 (TensorFlow & PyTorch)

모델 저장 및 로드 방법 (TensorFlow & PyTorch)딥러닝 모델을 학습하고 난 후 이를 저장하고 불러오는 과정은 매우 중요합니다. 학습된 모델을 재사용하거나 배포할 때 필요한 과정이며, 훈련을 중단한 후에도 모델을 다시 불러와 이어서 학습할 수 있습니다. 본 포스팅에서는 TensorFlow와 PyTorch에서 모델을 저장하고 로드하는 방법을 설명하겠습니다.1. TensorFlow에서 모델 저장 및 로드TensorFlow에서는 tf.keras.Model을 이용하여 모델을 저장할 수 있습니다. TensorFlow에서 제공하는 모델 저장 방법은 크게 두 가지로 나뉩니다.1.1 전체 모델 저장 및 로드TensorFlow에서는 model.save()를 사용하여 전체 모델을 저장할 수 있습니다. 이 방..

TensorFlow란? 설치 및 기본 사용법

TensorFlow란? 설치 및 기본 사용법1. TensorFlow란?TensorFlow는 Google Brain 팀에서 개발한 오픈 소스 딥러닝 프레임워크로, 머신러닝 및 딥러닝 모델을 구축하고 훈련하는 데 널리 사용됩니다. 다양한 플랫폼에서 실행할 수 있으며, GPU 및 TPU 가속을 활용하여 높은 성능을 제공합니다.TensorFlow의 특징다양한 플랫폼 지원: Windows, Linux, macOS뿐만 아니라 모바일 및 임베디드 기기에서도 실행 가능자동 미분 기능: 그래디언트 계산 자동화 (TensorFlow의 tf.GradientTape 활용)확장성: 작은 모델부터 대규모 분산 학습까지 지원풍부한 생태계: Keras API 통합, TensorFlow Lite, TensorFlow.js, Tenso..

반응형