모델저장
[부스트캠프][WK03 / Day13] PyTorch 구조 학습하기 2
1. 강의 내용 모델 불러오기 (최성철 교수님) 1) model.save() 학습의 결과를 저장하기 위한 함수 모델 형태(architecture)와 파라미터를 저장 모델 학습 중간 과정의 저장을 통해 최선의 결과모델을 선택 만들어진 모델을 외부 연구자와 공유하여 학습 재연성 향상 # state_dict: 모델의 파라메터를 표시 print("Model's state_dict:") for param_tensor in model.state_dict(): print(param_tensor, "\t", model.state_dict()[param_tensor].size()) # 모델의 파라메터를 저장 torch.save(model.state_dict(), os.path.join(MODEL_PATH, "model...