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

fixed default labels for QA model (#8399)

Browse Source
This commit is contained in:
Manav Rathod
2020-11-08 09:08:14 -05:00
committed by GitHub
parent 0b02489b2c
commit 610730998f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/trainer.py
View File

@@ -328,7 +328,7 @@ class Trainer:
self.hp_search_backend = None self.hp_search_backend = None
self.use_tune_checkpoints = False self.use_tune_checkpoints = False
default_label_names = ( default_label_names = (
["start_positions, end_positions"] ["start_positions", "end_positions"]
if type(self.model) in MODEL_FOR_QUESTION_ANSWERING_MAPPING.values() if type(self.model) in MODEL_FOR_QUESTION_ANSWERING_MAPPING.values()
else ["labels"] else ["labels"]
) )