Update run_glue.py

add DistilBert model shortcut into ALL_MODELS

Update run_glue.py
add DistilBert model shortcut into ALL_MODELS
7af07779 · Brian Ma · Lysandre Debut · c1689ac3 · 7af07779
Commit 7af07779 authored Oct 03, 2019 by Brian Ma Committed by Lysandre Debut Oct 03, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

examples/run_glue.py examples/run_glue.py +2 -1

No files found.
--- a/examples/run_glue.py
+++ b/examples/run_glue.py
@@ -53,7 +53,8 @@ from transformers import glue_convert_examples_to_features as convert_examples_t

 logger = logging.getLogger(__name__)

-ALL_MODELS = sum((tuple(conf.pretrained_config_archive_map.keys()) for conf in (BertConfig, XLNetConfig, XLMConfig, RobertaConfig)), ())
+ALL_MODELS = sum((tuple(conf.pretrained_config_archive_map.keys()) for conf in (BertConfig, XLNetConfig, XLMConfig, 
+                                                                                RobertaConfig, DistilBertConfig)), ())

 MODEL_CLASSES = {
    'bert': (BertConfig, BertForSequenceClassification, BertTokenizer),