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,
)
이렇게 사용하고 있습니다.
