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하기 전에 반드시 컴파일하도록 한다.