[Keras] model.save 에서 class로 만들어진 모델 load 오류

model 이 다른 class에서 만들어졌을 경우

model.save("modelname.h5")로 저장하면 로드가 안됨 (class 이름을 찾을 수 없다고 나옴)

model.save_weights("modelname.h5")로 저장

불러올 때는

model.load_weights("modelname.h5")로 불러옴

이 때 model은 기존에 저장할 때 사용했던 weight의 구조와 동일한 구조를 가지고 있어야 함

(save 할때 model이 layer 3층이었다면 동일한 크기의 weight를 가진 3층짜리 model을 만들어놓고 load_weights 해야 함)

 (저장 할 model 구조)

(load_weights 하는 예시)

참조: https://stackoverflow.com/questions/51806852/cant-save-custom-subclassed-model