소연의_개발일지
article thumbnail

 

CNN 알고리즘을 적용시켜 모델을 학습시키고 .save() 메서드를 사용하여 모델을 저장하는 과정 오류가 생겼다.

 

 

이 부분에서 오류가 생겼다.

 

오류 내용

UserWarning: You are saving your model as an HDF5 file via `model.save()`. This file format is considered legacy. We recommend using instead the native Keras format, e.g. `model.save('my_model.keras')`.
  saving_api.save_model(

이 경고 메세지는 Keras에서 모델을 저장할 때 HDF5 형식을 사용하는 것에 대한 경고이다.

 

이 경고의 주요 내용은

  • model.save() 메서드를 사용하여 모델을 HDF5 형식으로 저장하려고 합니다.
  • HDF5 형식은 이제 Keras에서 고려되는 레거시(legacy) 파일 형식입니다.
  • 대신 Keras의 네이티브 형식으로 모델을 저장하는 것이 권장됩니다.

 

오류가 뜨긴 하지만 문제없이 모델은 h5 형식으로 저장되는 걸 확인할 수 있다. 

 

 

해결 방법

1. 파일 확장자를 .h5 또는 .hdf5 대신 .keras로 사용하여 모델을 저장한다.

model.save('my_model.keras')

 

2. 위의 방법으로 저장된 모듈은 나중에 load_model 함수를 사용하여 로드할 수 있다.

from tensorflow.keras.models import load_model
loaded_model = load_model('my_model.keras')

 

위 방식대로 했더니 학습 모드를 실행시켜도 오류가 나오지 않았다.

또한 모델이 아래와 같이 잘 저장되었다.

profile

소연의_개발일지

@ssoyxon

포스팅이 좋았다면 "좋아요❤️" 또는 "구독👍🏻" 해주세요!