티스토리 뷰

deep learning/keras

keras load_model error

hellownd 2019. 6. 3. 21:30

keras load_model에서 compile 에러 발생 시 해결 방안

 

- 케라스모델은 저장할 때 컴파일 이후에 저장해야한다!

 

케라스 모델을 컴파일하지 않고 저장해도 자신의 컴퓨터에서는 로드가 될 수 있지만

저장한 모델 파일을 다른 컴퓨터로 옮긴 뒤 실행했을 때 문제가 발생할 수 있다.

 

model = Model(...)

model.compile(..)

model.save("model.h5")

 

모델을 save하기 전에 반드시 컴파일하도록 한다.

 

댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2025/04   »
1 2 3 4 5
6 7 8 9 10 11 12
13 14 15 16 17 18 19
20 21 22 23 24 25 26
27 28 29 30
글 보관함