더북(TheBook)

Note ≣ | model.save() 에러

모델 구조와 설정, 추가로 구성한 레이어까지 한 번에 저장하려면 model.save() 함수를 사용하면 되는데, 위 예제에서는 굳이 model.save_weights() 함수를 사용했습니다. 그 이유는 transformers 라이브러리를 통해 로드된 TF 모델(정확히는 tf.keras)에 모델의 전체 구조를 저장할 때 필요한 configuration이 포함되어 있지 않아 model.save()를 실행할 때 에러가 발생하기 때문입니다. 따라서 저장할 때는 모델 가중치만 저장하는 save_weights()를 사용하고, 나중에 추론할 때는 모델 구조를 다시 불러와서 가중치를 모델에 입히는 방식으로 진행해야 합니다.13

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.