6. 모델 저장, 불러오기

저장 : torch.save(model_name.state_dict(), path)

- checkpoints : earlystopping 사용 시 중간중간 최선의 결과를 저장하는 것.

- MLops 사용하는게 어떻나.. 생각해봄.

 

불러오기 : model = torch.load(path)

- transfer learning 수행.

- backbone architecture가 잘 학습된 모델에서 일부분만 변경하여 학습을 수행하는 것

- 주로 대용량 데이터로 학습된 모델을 불러오기 때문에 성능이 괜찮다.

 

요즘은 transfer learning에서 stacking frozen 학습 방식도 사용한다.

- frozen or train : layer.requires_grad = False 지정.