Skip to content

GitLab

"web/git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "de2894ac2cdcf0ece2c0d2a576579768882e8882"
Commit 852e032c authored by VictorSanh's avatar VictorSanh
Browse files

include roberta in run_squad_w_distillation - cc @graviraja

parent b5509abb
Hide 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