본문 바로가기

Deep Learning

Tensorflow의 모델을 save, load하는 방법

학습시킨 모델을 폴더로 Save하는 방법

 

위와같이 모델을 저장할 경로를 지정하고

미리 만들어둔 모델의 변수명 뒤에 .save를 사용하여 원하는 파일이름으로 저장 할 수 있다.

 

 

저장된 파일을 load할 때에는

텐서플로우의 load_model 함수를 사용하여 불러 올 수 있다.

 

위의 방법들은 학습시킨 모델을 전부 저장하고 불러오는 방법이다.

 

 

 

학습시킨 모델을 File로 저장하는 방법