파이토치에서는 어떻게 모델을 저장하고 불러오나요?

[9bow (박정환)]

공식 홈페이지와 StackOverflow 등에서 자주 보이는 질문과 답변을 번역하고 있습니다.

다음 링크에서 원문을 함께 찾아보실 수 있습니다.

• 원문: N/A

---

질문

• 학습을 마친 뒤 파일에 저장하고 다시 불러와서 학습을 계속하려면 어떻게 해야 하나요?

[9bow (박정환)]

답변

• 모델을 저장하고 불러오는 몇 가지 방법이 있습니다. 아래 튜토리얼을 참고해보세요.
  모델 저장하고 불러오기 — PyTorch Tutorials 1.11.0+cu102 documentation

• 아래 링크의 레시피들도 참고해보실 수 있습니다.
  PyTorch에서 여러 모델을 하나의 파일에 저장하기 & 불러오기 — PyTorch Tutorials 1.11.0+cu102 documentation
  PyTorch에서 일반적인 체크포인트(checkpoint) 저장하기 & 불러오기 — PyTorch Tutorials 1.11.0+cu102 documentation
  PyTorch에서 추론(inference)을 위해 모델 저장하기 & 불러오기 — PyTorch Tutorials 1.11.0+cu102 documentation
  PyTorch에서 다양한 장치 간 모델을 저장하고 불러오기 — PyTorch Tutorials 1.11.0+cu102 documentation