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

[config] Ensure that id2label always takes precedence over num_labels

Browse Source 
Fixes bug reported in https://github.com/huggingface/transformers/issues/4669

See #3967 for context
This commit is contained in:
Julien Chaumond
2020-06-01 16:25:43 +02:00
parent ec62b7d953
commit 751a1e0890
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/configuration_utils.py
View File

@@ -89,6 +89,7 @@ class PretrainedConfig(object):
self.id2label = kwargs.pop("id2label", None) self.id2label = kwargs.pop("id2label", None)
self.label2id = kwargs.pop("label2id", None) self.label2id = kwargs.pop("label2id", None)
if self.id2label is not None: if self.id2label is not None:
del kwargs["num_labels"]
self.id2label = dict((int(key), value) for key, value in self.id2label.items()) self.id2label = dict((int(key), value) for key, value in self.id2label.items())
# Keys are always strings in JSON so convert ids to int here. # Keys are always strings in JSON so convert ids to int here.
else: else: