You passed along `num_labels=3` with an incompatible id to label map: {'0': 'NEGATIVE', '1': 'POSITIVE'}. The number of labels wil be overwritten to 2.
#2
by
dfsfb5r64
- opened
num_labels을 지정하는것과 관계없이 저 문구가 계속 뜨는데.. 제가 잘못하고 있는 부분이 있는건가용?
tokenizer = AutoTokenizer.from_pretrained('hyunwoongko/kobart')
model = AutoModelForSequenceClassification.from_pretrained(
pretrained_model_name_or_path = 'hyunwoongko/kobart',
num_labels = 564,
id2label = idx_to_SS,
label2id = SS_to_idx,
)
이렇게 사용하고 있습니다.
저도 같은 증상이 반복적으로 나타납니다.
혹시 해결되셨을까요?
@dfsfb5r64