Tensoflow를 사용하다보면, 모델을 생성하고, 학습을 하게 된 후, 나중에 이를 다른 기기나 서버 등으로 옮겨서 배포하게 된다. 이때 모델 과 학습이 끝난 weight(가중치)를 파일로 저장하고 불러올 수 있는 방법이다.
0. 전체 모델과 웨이트를 h5파일로 통째로 저장하고 불러오기
1. 전체 모델과 웨이트를 폴더구조로 통째로 편하게 처리 : ( 이때는 컴파일을 할 필요 없이 바로 사용 가능! )
2. 네트워크의 구조(모델) 만 저장하고 불러오기
3. 네트워크의 가중치를 저장하고 불러오기 : 이때는 컴파일을 꼭 해줘야 사용이 가능함!
4. 스케일러 등을 저장하고 불러오기 : joblib 라이브러리를 이용한다!
반응형
'AI 이론 > 딥러닝 이론' 카테고리의 다른 글
| [딥러닝] 이미지 분류 코딩 과정 요약/ astype(), to_categorical, argmax(axis = 1) (0) | 2021.12.01 |
|---|---|
| 선형대수학 참고 동영상 (0) | 2021.12.01 |
| [모델링함수만들기] (0) | 2021.12.01 |
| [이미지 분류 모델링] Flatten(), softmax, evaluate(), sparse_categorical_crossentropy (0) | 2021.12.01 |
| [콜백 2가지 방법] callbacks.EarlyStopping(), class myCallback(tf.keras.callbacks.Callback) (0) | 2021.11.30 |