슬기로운 연구생활

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

슬기로운 에러 생활

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)

'슬기로운 에러 생활' 카테고리의 다른 글

git fatal out of memory malloc failed  (0) 2021.09.13
couldn't communicate with the NVIDIA Driver  (0) 2021.09.13
[bentoml] module not found error: no module named  (1) 2021.08.30
AttributeError: 'str' object has no attribute 'decode'  (0) 2021.08.26
NotImplementedError: Cannot convert a symbolic Tensor (lstm/strided_slice:0) to a numpy array.  (0) 2021.08.26
'슬기로운 에러 생활' Related Articles more
Comments