Skip to content

GitLab

Unverified Commit 82a2b76c authored by Koichi Yasuoka's avatar Koichi Yasuoka Committed by GitHub
Browse files

Update run_ner.py with id2label config (#12001)

parent 0e82f0cb
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
examples/pytorch/token-classification/run_ner.py
View file @ 82a2b76c
......@@ -281,6 +281,8 @@ def main():
config = AutoConfig.from_pretrained(
model_args.config_name if model_args.config_name else model_args.model_name_or_path,
num_labels=num_labels,
label2id=label_to_id,
id2label={i: l for l, i in label_to_id.items()},
finetuning_task=data_args.task_name,
cache_dir=model_args.cache_dir,
revision=model_args.model_revision,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment