RuntimeError: The size of tensor a (3902) must match the size of tensor b (512) at non-singleton dimension 1

우주먼지·2021년 11월 29일
0

KoBert fine-turning을 통해 사용하는 중에 다음과 같은 에러를 만났다.

이 에러는 pre-trained 모델에서 사용했던 max-len보다 긴 데이터가 입력으로 들어올 경우 발생한 에러이다.

해결방법 또한 간단하다.

그저 최대 길이를 맞춰주면 된다.

위와 같은 에러 문구가 발생했다면, a는 입력 문구의 길이이고, b는 모델의 최대 길이이다.

따라서 a를 b보다 작은 수가 될때만 입력하도록 하면 된다.

profile
안녕하세요 ㅎㅎ

0개의 댓글