[에러] Please ensure this object is passed to the `custom_objects` argument.

딥러닝 모델을 학습할 때 항상 너무 답답한건 바로 학습 속도이다.

또한 코랩을 사용하는 유저라면 학습데이터가 크거나 모델의 복잡도가 높을 수록 툭하면 램 사용량이 터져버리는 상황이 발생한다.

이런 상황을 대비하기 위해 어렵게 학습한 모델을 로컬 저장소에 저장하는 방법을 많이 쓴다.

 

나도 역시나 과제를 하다가 틈만 나면 코랩이 터져버리는 현상 때문에 모델을 저장하고 다음 날 모델을 불러왔는데,다음과 같은 에러가 발생하였다.

Unknown metric function: psnr_metric. Please ensure this object is passed to the `custom_objects` argument. 

에러문을 해석해보면,

내가 커스텀한 모델과 매트릭스를 불러 올 때, 에러가 생기는 모양이다.

 

내가 커스텀한 모델은 아래와 같다.

 

오랫동안 학습한 모델이 다 날라갔나 해서 식은땀이 좔좔 흘렀는데 구글링을 통해 해결방법을 알아냈다.

 

from tensorflow.keras.models import load_model
model=load_model('model9-4.h5', compile=False)

load_model 안에서 compile=False 이 부분을 추가해주면 된다.

 

너무나 간단해서 코쓱머쓱,,, 

 

그치만 해결해서 넘나 다행 ㅎㅎㅎ