xswer19.log
로그인
xswer19.log
로그인
Model_save_and_Loda
안선경
·
2023년 4월 26일
팔로우
0
0
DL_study
목록 보기
9/23
model을 구축하고 해당 model를 저장해서 필요할 때 사용할 수 있다.
먼저 save함수를 사용해서 모델 및 weight값을 저장할 수 있다.
중요한 점은 끝에 .h5 넣어줌으로써 파일확장값을 지정해야한다.
불러올 때는 tensorflow.keras.model.load_model('파일명.h5)라고 넣어줘야한다.
save_weights함수는 가중치만을 저장하는데 저장 공간을 절약할 수 있다.
그래서 불러올 때 먼저 모델을 구축하고, 구축 후 load_weights만 이용해서 모델을 불러올 수 있다.
callback함수를 사용해서 불러올 수도 있다.
tensorflow.callback모듈에 체크포인트를 불러올 수 있는데, 변수로 저장하고 model.fit에서 callback입력값에 넣어주면 된다.
이번에는 파일 확장자명이 다른 pb형식으로 저장해봤다.
해당 파일은 .5h로 저장되지 않으며, 불러오는 방법은 똑같다.
안선경
상황을 바꿀 수 없다면, 나를 바꾸자
팔로우
이전 포스트
Tensorboard
다음 포스트
Data
0개의 댓글
댓글 작성