반응형
Tensoflow를 사용하다보면, 모델을 생성하고, 학습을 하게 된 후, 나중에 이를 다른 기기나 서버 등으로 옮겨서 배포하게 된다. 이때 모델 과 학습이 끝난 weight(가중치)를 파일로 저장하고 불러올 수 있는 방법이다.
0. 전체 모델과 웨이트를 h5파일로 통째로 저장하고 불러오기
1. 전체 모델과 웨이트를 폴더구조로 통째로 편하게 처리 : ( 이때는 컴파일을 할 필요 없이 바로 사용 가능! )
2. 네트워크의 구조(모델) 만 저장하고 불러오기
3. 네트워크의 가중치를 저장하고 불러오기 : 이때는 컴파일을 꼭 해줘야 사용이 가능함!
4. 스케일러 등을 저장하고 불러오기 : joblib 라이브러리를 이용한다!
반응형
'DeepLearining' 카테고리의 다른 글
| 텐서플로우 딥러닝 GridSearchCV 이용 방법 (1) | 2021.03.12 |
|---|---|
| Learning Curve - 텐서플로우 딥러닝 (0) | 2021.03.12 |
| 옵티마이저 (Optimizer) 종류 - 인공지능, 머신러닝, 데이터마이닝 (0) | 2020.12.10 |
| Label Encoding and One Hot Encoding (0) | 2017.10.29 |
| 딥러닝의 모델 성능 평가2 - manual 로 varification dataset 만들기 (0) | 2017.10.29 |