Skip to content

GitLab

Commit 189ff9b6 authored by Christophe Bourguignat's avatar Christophe Bourguignat Committed by Lysandre Debut
Browse files

Update README after RoBERTa addition

parent d8923270
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
README.md
View file @ 189ff9b6
...@@ -76,7 +76,7 @@ import torch ...@@ -76,7 +76,7 @@ import torch
from pytorch_transformers import * from pytorch_transformers import *
# PyTorch-Transformers has a unified API # PyTorch-Transformers has a unified API
# for 6 transformer architectures and 27 pretrained weights. # for 7 transformer architectures and 30 pretrained weights.
# Model | Tokenizer | Pretrained weights shortcut # Model | Tokenizer | Pretrained weights shortcut
MODELS = [(BertModel, BertTokenizer, 'bert-base-uncased'), MODELS = [(BertModel, BertTokenizer, 'bert-base-uncased'),
(OpenAIGPTModel, OpenAIGPTTokenizer, 'openai-gpt'), (OpenAIGPTModel, OpenAIGPTTokenizer, 'openai-gpt'),
......
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