목록PyTorch Model save (1)
호기심 많은 분석가

부스트캠프 개발자의 지속 가능한 성장을 위한 학습 커뮤니티 boostcamp.connect.or.kr 개인학습 (06강) 모델 불러오기 💡 PyTorch 모델 저장 및 불러오는 방법과 이어 학습하기(Transfer Learning)를 학습 요즘 딥러닝은 backbone, 즉 이미 학습이 되어있는 모델을 가져와서 우리 데이터에 맞춰서 다시 학습하는 fine tuning이 대세다. 이미지에서는 ResNet 계열들, NLP에서는 Bert 계열 모델들 우리는 학습 결과를 공유하고 싶다. 어떻게? Tip. Colab에서 작업 끝나면 Slack으로 메세지 보내기 1. model.save() 학습의 결과를 저장하기 위한 함수 모델 형태 (architecture)와 파라미터를 저장 모델 학습 중간 과정의 저장을 통해 ..
Coding/BoostCamp
2021. 8. 21. 20:15