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