Model Checkpointing — 학습 중간 상태를 저장·복원하다

Model Checkpointing

학습이 오래 걸리는 만큼, 중간 상태(모델·옵티마이저·에폭·손실)를 디스크에 저장해 문제가 생겨도 이어 갈 수 있어야 해요. Model Checkpointing(모델 체크포인트) 은 이런 상태를 저장(save)하고 복원(load)하는 기법이에요. PyTorch의 torch.save·torch.load·state_dict가 대표적이에요.

이 카테고리의 문서

• 개요: PyTorch 튜토리얼 — 모델 저장·불러오기(체크포인트)
• 핵심 기능: torch.save — 직렬화 저장 API
• 실전·API: torch.load — 역직렬화·로드 API

출처: https://pytorch.org/tutorials/beginner/saving_loading_models.html