[Tensorflow] model.save() 오류

model.save('파일이름.h5')로 했을 때 오류가 났었다.

 

해당 오류의 경우

model.save_weights('파일이름.h5')

이렇게 저장하여 오류가 해결됐다.

 

save, load_model은 모델 전체를 저장하기 때문에, load 이후에 별도로 처리할 필요가 없다.

 

save_weights, load_weights는 가중치만 저장하기 때문에, 모델 architecture를 동일하게 만들어줘야 된다.

이미 모델 architecture를 알고 있을때만 사용할 수 있다.

 

참고

https://ltlkodae.tistory.com/13