From 9d0029e215f5ad0836d6be87458aab5142783af4 Mon Sep 17 00:00:00 2001 From: LysandreJik Date: Thu, 15 Aug 2019 17:17:35 -0400 Subject: [PATCH] Added RoBERTa example to README --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 389c2f25ad..3389e10593 100644 --- a/README.md +++ b/README.md @@ -83,7 +83,8 @@ MODELS = [(BertModel, BertTokenizer, 'bert-base-uncased'), (GPT2Model, GPT2Tokenizer, 'gpt2'), (TransfoXLModel, TransfoXLTokenizer, 'transfo-xl-wt103'), (XLNetModel, XLNetTokenizer, 'xlnet-base-cased'), - (XLMModel, XLMTokenizer, 'xlm-mlm-enfr-1024')] + (XLMModel, XLMTokenizer, 'xlm-mlm-enfr-1024'), + (RobertaModel, RobertaTokenizer, 'roberta-base')] # Let's encode some text in a sequence of hidden-states using each model: for model_class, tokenizer_class, pretrained_weights in MODELS: