[PLM] IndexError: Target X is out of bounds.

PyTorch+KoELECTRA 분류 모델 사용 중에 뜬 에러 문구이다.

 

1. 모델을 불러온 방식(에러 발생)

분류를 위한 KoELECTRA 모델을 사용했는데 처음에는 필요한 레이블 수를 지정해주지 않았다.

 

2. 수정

분류해야 하는 레이블 수만큼 num_labels 인자로 추가해주었다. kcBERT 깃허브에서 똑같은 질문을 발견했는데 개발자 분께서 ~ForSequenceClassification 모델은 기본이 이진 분류라고 답변해주신 내용을 참고했다. KoELECTRA 모델도 BERT 기반이니까 저 방법이 될수도 있겠다 싶었고 적용하니까 에러 해결 🙂

 

참고 사이트

https://github.com/Beomi/KcBERT/issues/3