Model_save_and_Loda

안선경·2023년 4월 26일
0

DL_study

목록 보기
9/23
post-thumbnail
post-custom-banner

  • model을 구축하고 해당 model를 저장해서 필요할 때 사용할 수 있다.
  • 먼저 save함수를 사용해서 모델 및 weight값을 저장할 수 있다.
  • 중요한 점은 끝에 .h5 넣어줌으로써 파일확장값을 지정해야한다.
  • 불러올 때는 tensorflow.keras.model.load_model('파일명.h5)라고 넣어줘야한다.
  • save_weights함수는 가중치만을 저장하는데 저장 공간을 절약할 수 있다.
  • 그래서 불러올 때 먼저 모델을 구축하고, 구축 후 load_weights만 이용해서 모델을 불러올 수 있다.
  • callback함수를 사용해서 불러올 수도 있다.
  • tensorflow.callback모듈에 체크포인트를 불러올 수 있는데, 변수로 저장하고 model.fit에서 callback입력값에 넣어주면 된다.
  • 이번에는 파일 확장자명이 다른 pb형식으로 저장해봤다.
  • 해당 파일은 .5h로 저장되지 않으며, 불러오는 방법은 똑같다.
profile
상황을 바꿀 수 없다면, 나를 바꾸자
post-custom-banner

0개의 댓글