pytorch checkpoints 썸네일형 리스트형 [pytorch] 학습 결과 및 모델 저장하기 model.save() 학습 결과 저장 함수 모델 형태(architecture)와 파라미터를 저장할 수 있음 모델 학습 중간 과정을 저장함으로써 최선의 결과를 가지는 모델을 선택할 수 있음 만들어진 모델을 외부 연구자와 공유해 학습 재연성을 향상할 수 있음 state_dict : 모델의 파라미터를 의미 # 모델의 파라미터를 저장 torch.save(model.state_dict(), os.path.join(MODEL_PATH, "model.pt")) # 동일한 형태의 모델을 만들어 파라미터만 로드하여 사용 new_model = TheModelClass() new_model.load_state_dict(torch.load(os.path.join(MODEL_PATH, "model.pt"))) # 모델 arc.. 더보기 이전 1 다음