슬기로운 에러 생활

Missing key(s) in state_dict: "embeddings.position_ids".

vhrehfdl 2021. 9. 9. 08:03

* 에러 원인 : kobert를 사용하다 발생했다. 미리 학습된 모델의 weight 값을 load_state_dict를 사용했다. 여기서 기존 버전과 모델이 맞지 않아 발생한 에러인 것 같다.

* 해결 방법 : load_state_dict에서 strict를 False를 설정하면 해결된다.

bertmodel.load_state_dict(torch.load(model_file), strict=False)