SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update label2id in the model config for run_glue (#13334)

Browse Source
This commit is contained in:
Sylvain Gugger
2021-08-30 10:35:09 -04:00
committed by GitHub
parent 6f3c99acca
commit 139e830158
3 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/pytorch/text-classification/run_glue.py
View File

@@ -380,6 +380,9 @@ def main():
if label_to_id is not None: if label_to_id is not None:
model.config.label2id = label_to_id model.config.label2id = label_to_id
model.config.id2label = {id: label for label, id in config.label2id.items()} model.config.id2label = {id: label for label, id in config.label2id.items()}
elif data_args.task_name is not None and not is_regression:
model.config.label2id = {l: i for i, l in enumerate(label_list)}
model.config.id2label = {id: label for label, id in config.label2id.items()}
if data_args.max_seq_length > tokenizer.model_max_length: if data_args.max_seq_length > tokenizer.model_max_length:
logger.warning( logger.warning(

3
examples/pytorch/text-classification/run_glue_no_trainer.py
View File

@@ -288,6 +288,9 @@ def main():
if label_to_id is not None: if label_to_id is not None:
model.config.label2id = label_to_id model.config.label2id = label_to_id
model.config.id2label = {id: label for label, id in config.label2id.items()} model.config.id2label = {id: label for label, id in config.label2id.items()}
elif args.task_name is not None and not is_regression:
model.config.label2id = {l: i for i, l in enumerate(label_list)}
model.config.id2label = {id: label for label, id in config.label2id.items()}
padding = "max_length" if args.pad_to_max_length else False padding = "max_length" if args.pad_to_max_length else False

3
examples/tensorflow/text-classification/run_glue.py
View File

@@ -355,6 +355,9 @@ def main():
if label_to_id is not None: if label_to_id is not None:
config.label2id = label_to_id config.label2id = label_to_id
config.id2label = {id: label for label, id in config.label2id.items()} config.id2label = {id: label for label, id in config.label2id.items()}
elif data_args.task_name is not None and not is_regression:
config.label2id = {l: i for i, l in enumerate(label_list)}
config.id2label = {id: label for label, id in config.label2id.items()}
if data_args.max_seq_length > tokenizer.model_max_length: if data_args.max_seq_length > tokenizer.model_max_length:
logger.warning( logger.warning(