Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "927904bc911225c1ea6087c5e24aa77d265bfb9e"
Commit 852e032c authored by VictorSanh's avatar VictorSanh
Browse files

include roberta in run_squad_w_distillation - cc @graviraja

parent b5509abb
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
examples/distillation/run_squad_w_distillation.py
View file @ 852e032c
...@@ -45,6 +45,9 @@ from transformers import ( ...@@ -45,6 +45,9 @@ from transformers import (
XLNetConfig, XLNetConfig,
XLNetForQuestionAnswering, XLNetForQuestionAnswering,
XLNetTokenizer, XLNetTokenizer,
RobertaConfig,
RobertaForQuestionAnswering,
RobertaTokenizer,
get_linear_schedule_with_warmup, get_linear_schedule_with_warmup,
squad_convert_examples_to_features, squad_convert_examples_to_features,
) )
...@@ -73,6 +76,7 @@ MODEL_CLASSES = { ...@@ -73,6 +76,7 @@ MODEL_CLASSES = {
"xlnet": (XLNetConfig, XLNetForQuestionAnswering, XLNetTokenizer), "xlnet": (XLNetConfig, XLNetForQuestionAnswering, XLNetTokenizer),
"xlm": (XLMConfig, XLMForQuestionAnswering, XLMTokenizer), "xlm": (XLMConfig, XLMForQuestionAnswering, XLMTokenizer),
"distilbert": (DistilBertConfig, DistilBertForQuestionAnswering, DistilBertTokenizer), "distilbert": (DistilBertConfig, DistilBertForQuestionAnswering, DistilBertTokenizer),
"roberta": (RobertaConfig, RobertaForQuestionAnswering, RobertaTokenizer),
} }
......
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