TorchServe 맛보기

TorchServe 맛보기 TorchServe를 간단히 시작하기 위한 정리입니다. pytorch/serve에서 제공해주는 학습 모델을 사용하여 RestApi로 추론하는 방법입니다. Torchserve는 Docker 이미지를 사용합니다. 참고 https://github.com/pytorch/serve/tree/master/examples/image_classifier/densenet_161 모델압축 모델을 학습하게 되면 로컬에 pth 파일이 생성이 됩니다. 이 모델 파일을 그대로 TorchServe에서는 사용 할 수 없기 때문에 먼저 Training이 끝난 모델을 압축해야 합니다. densenet161 학습 모델 다운로드 model.py TorchServe에서 제공해주는 모델을 다운로드 받습니다. 저는 D..

2023.04.19