모델 저장 및 불러오기

한 제조 회사가 이미지 기반으로 프로젝트를 분류해 적절한 배송 포장을 결정하려고 합니다. PyTorch로 높은 정확도의 모델을 학습했으므로, 이제 모델과 사전 학습된 가중치를 저장해 두고 팀과 공유하여 누구나 끊김 없이 불러올 수 있도록 하려 합니다.

torch와 torch.nn(nn)이 임포트되어 있어요. 사전 학습된 모델 객체는 워크스페이스에서 model로, 모델 아키텍처는 ManufacturingCNN으로 제공됩니다.