[KLUE-STS] MSE loss를 사용한 학습방법이 궁금합니다.

[dongju923]

Description

안녕하세요. 데이터셋 공유에 정말 감사드립니다.
논문에서 "The model is thus trained to map from the final hidden state of [CLS] to a real number, by minimizing the mean squared error (MSE)." 이렇게 나와 있는데, [CLS] 에 대한 hidden_state를 뽑아도 (batch_size, embedding_dim) 크기가 반환될텐데, labels랑 차원을 어떻게 맞추셨는지 궁금합니다. 감사합니다.