fixed default labels for QA model (#8399)

This commit is contained in:
Manav Rathod
2020-11-08 09:08:14 -05:00
committed by GitHub
parent 0b02489b2c
commit 610730998f

View File

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