Sagemaker에서 PyTorch를 활용한 딥러닝 이미지 분류 모델 학습 및 배포

본 글은 AWS SageMaker를 활용하여 PyTorch로 딥러닝 이미지 분류 모델을 학습하고 배포하는 과정을 다룹니다. 특히, Computer Vision 분야에서 유용한 torchvision 패키지를 사용하여 사용자 정의 데이터셋으로 고양이 이미지 분류 모델을 구축하는 방법을 설명합니다.

데이터셋 구성부터 모델 학습, 그리고 Serverless API 개발까지 두 파트로 나누어 설명하며, 오늘은 딥러닝 모델 개발 부분인 SageMaker에서의 학습 및 배포 과정을 집중적으로 다룹니다. 이를 통해 딥러닝 모델 개발 및 클라우드 환경에서의 배포 과정을 이해하는 데 도움을 줄 것입니다.