[PyTroch] model.save()

hj choi·2022년 9월 29일
0

AI

목록 보기
11/27

model.save()

  • 학습 결과를 저장하기 위한 함수
  • 모델의 형태(architecture)와 파라메터를 저장
  • 모델 학습 중간 과정의 을 통해 최선의 결과모델을 선택(early stopping 등)
  • 만들어진 모델을 외부 연구자와 공유하여 학습 재연성 향상

  • 모델의 architecture와 함께 저장하는 방식은 pickle형태로 저장된다.

0개의 댓글