chjcoder.log
로그인
chjcoder.log
로그인
[PyTroch] model.save()
hj choi
·
2022년 9월 29일
팔로우
0
AI
PyTorch
python
0
AI
목록 보기
11/27
model.save()
학습 결과를 저장하기 위한 함수
모델의 형태(architecture)와 파라메터를 저장
모델 학습 중간 과정의 을 통해 최선의 결과모델을 선택(early stopping 등)
만들어진 모델을 외부 연구자와 공유하여 학습 재연성 향상
모델의 architecture와 함께 저장하는 방식은 pickle형태로 저장된다.
hj choi
팔로우
이전 포스트
[PyTorch] PyTorch datasets & dataloaders
다음 포스트
[PyTorch] 딥러닝 checkpoint 저장
0개의 댓글
댓글 작성