As traduções são geradas por tradução automática. Em caso de conflito entre o conteúdo da tradução e da versão original em inglês, a versão em inglês prevalecerá.
TensorFlow Modelos de hub
Os seguintes modelos pré-treinados estão disponíveis para uso no aprendizado por transferência com o TensorFlow algoritmo de Classificação de Texto.
Os modelos a seguir variam significativamente em tamanho, número de parâmetros do modelo, tempo de treinamento e latência de inferência para qualquer conjunto de dados. O melhor modelo para seu caso de uso depende da complexidade do seu conjunto de dados de ajuste fino e de quaisquer requisitos que você tenha sobre tempo de treinamento, latência de inferência ou precisão do modelo.
| Nome do modelo | model_id |
Fonte |
|---|---|---|
|
Base BERT uncased |
|
|
|
Base BERT cased |
|
|
|
Estojo multilíngue Base BERT |
|
|
|
BERT pequeno L-2 _ H-128 _ A-2 |
|
|
|
BERT pequeno L-2 _ H-256 _ A-4 |
|
|
|
BERT pequeno L-2 _ H-512 _ A-8 |
|
|
|
BERT pequeno L-2 _ H-768 _ A-12 |
|
|
|
BERT pequeno L-4 _ H-128 _ A-2 |
|
|
|
BERT pequeno L-4 _ H-256 _ A-4 |
|
|
|
BERT pequeno L-4 _ H-512 _ A-8 |
|
|
|
BERT pequeno L-4 _ H-768 _ A-12 |
|
|
|
BERT pequeno L-6 _ H-128 _ A-2 |
|
|
|
BERT pequeno L-6 _ H-256 _ A-4 |
|
|
|
BERT pequeno L-6 _ H-512 _ A-8 |
|
|
|
BERT pequeno L-6 _ H-768 _ A-12 |
|
|
|
BERT pequeno L-8 _ H-128 _ A-2 |
|
|
|
BERT pequeno L-8 _ H-256 _ A-4 |
|
|
|
BERT pequeno L-8 _ H-512 _ A-8 |
|
|
|
BERT pequeno L-8 _ H-768 _ A-12 |
|
|
|
BERT pequeno L-10 _ H-128 _ A-2 |
|
|
|
BERT pequeno L-10 _ H-256 _ A-4 |
|
|
|
BERT pequeno L-10 _ H-512 _ A-8 |
|
|
|
BERT pequeno L-10 _ H-768 _ A-12 |
|
|
|
BERT pequeno L-12 _ H-128 _ A-2 |
|
|
|
BERT pequeno L-12 _ H-256 _ A-4 |
|
|
|
BERT pequeno L-12 _ H-512 _ A-8 |
|
|
|
BERT pequeno L-12 _ H-768 _ A-12 |
|
|
|
BERT grande uncased |
|
|
|
BERT grande cased |
|
|
|
Máscara de palavras inteiras BERT grande uncased |
|
|
|
Máscara de palavras inteiras BERT grande cased |
|
|
|
Base ALBERT |
|
|
|
ELECTRA Small++ |
|
|
|
Base ELECTRA |
|
|
|
BERT Base Wikipedia e BooksCorpus |
|
|
|
Base BERT MEDLINE/PubMed |
|
|
|
Base Talking Heads |
|
|
|
Base Talking Heads |
|
