Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ed6c8d37f402847e65510a81e79b483c68733fe4"
Unverified Commit 61073099 authored by Manav Rathod's avatar Manav Rathod Committed by GitHub
Browse files

fixed default labels for QA model (#8399)

parent 0b02489b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/trainer.py
View file @ 61073099
...@@ -328,7 +328,7 @@ class Trainer: ...@@ -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"]
) )
......
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