슬기로운 에러 생활
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)