SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update README after RoBERTa addition

Browse Source
This commit is contained in:
Christophe Bourguignat
2019-08-17 18:46:50 +02:00
committed by Lysandre Debut
parent d8923270e6
commit 189ff9b664
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -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'),