Modelos do TensorFlow Hub - SageMaker IA da Amazon

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á.

Modelos do TensorFlow Hub

Os seguintes modelos pré-treinados estão disponíveis para uso no aprendizado por transferência com o algoritmo de classificação de texto - TensorFlow:

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 Origem

Base BERT uncased

tensorflow-tc-bert-en-uncased-L-12-H-768-A-12-2

Link do TensorFlow Hub

Base BERT cased

tensorflow-tc-bert-en-cased-L-12-H-768-A-12-2

Link do TensorFlow Hub

Estojo multilíngue Base BERT

tensorflow-tc-bert-multi-cased-L-12-H-768-A-12-2

Link do TensorFlow Hub

BERT pequeno L-2_H-128_A-2

tensorflow-tc-small-bert-bert-en-uncased-L-2-H-128-A-2

Link do TensorFlow Hub

BERT pequeno L-2_H-256_A-4

tensorflow-tc-small-bert-bert-en-uncased-L-2-H-256-A-4

Link do TensorFlow Hub

BERT pequeno L-2_H-512_A-8

tensorflow-tc-small-bert-bert-en-uncased-L-2-H-512-A-8

Link do TensorFlow Hub

BERT pequeno L-2_H-768_A-12

tensorflow-tc-small-bert-bert-en-uncased-L-2-H-768-A-12

Link do TensorFlow Hub

BERT pequeno L-4_H-128_A-2

tensorflow-tc-small-bert-bert-en-uncased-L-4-H-128-A-2

Link do TensorFlow Hub

BERT pequeno L-4_H-256_A-4

tensorflow-tc-small-bert-bert-en-uncased-L-4-H-256-A-4

Link do TensorFlow Hub

BERT pequeno L-4_H-512_A-8

tensorflow-tc-small-bert-bert-en-uncased-L-4-H-512-A-8

Link do TensorFlow Hub

BERT pequeno L-4_H-768_A-12

tensorflow-tc-small-bert-bert-en-uncased-L-4-H-768-A-12

Link do TensorFlow Hub

BERT pequeno L-6_H-128_A-2

tensorflow-tc-small-bert-bert-en-uncased-L-6-H-128-A-2

Link do TensorFlow Hub

BERT pequeno L-6_H-256_A-4

tensorflow-tc-small-bert-bert-en-uncased-L-6-H-256-A-4

Link do TensorFlow Hub

BERT pequeno L-6_H-512_A-8

tensorflow-tc-small-bert-bert-en-uncased-L-6-H-512-A-8

Link do TensorFlow Hub

BERT pequeno L-6_H-768_A-12

tensorflow-tc-small-bert-bert-en-uncased-L-6-H-768-A-12

Link do TensorFlow Hub

BERT pequeno L-8_H-128_A-2

tensorflow-tc-small-bert-bert-en-uncased-L-8-H-128-A-2

Link do TensorFlow Hub

BERT pequeno L-8_H-256_A-4

tensorflow-tc-small-bert-bert-en-uncased-L-8-H-256-A-4

Link do TensorFlow Hub

BERT pequeno L-8_H-512_A-8

tensorflow-tc-small-bert-bert-en-uncased-L-8-H-512-A-8

Link do TensorFlow Hub

BERT pequeno L-8_H-768_A-12

tensorflow-tc-small-bert-bert-en-uncased-L-8-H-768-A-12

Link do TensorFlow Hub

BERT pequeno L-10_H-128_A-2

tensorflow-tc-small-bert-bert-en-uncased-L-10-H-128-A-2

Link do TensorFlow Hub

BERT pequeno L-10_H-256_A-4

tensorflow-tc-small-bert-bert-en-uncased-L-10-H-256-A-4

Link do TensorFlow Hub

BERT pequeno L-10_H-512_A-8

tensorflow-tc-small-bert-bert-en-uncased-L-10-H-512-A-8

Link do TensorFlow Hub

BERT pequeno L-10_H-768_A-12

tensorflow-tc-small-bert-bert-en-uncased-L-10-H-768-A-12

Link do TensorFlow Hub

BERT pequeno L-12_H-128_A-2

tensorflow-tc-small-bert-bert-en-uncased-L-12-H-128-A-2

Link do TensorFlow Hub

BERT pequeno L-12_H-256_A-4

tensorflow-tc-small-bert-bert-en-uncased-L-12-H-256-A-4

Link do TensorFlow Hub

BERT pequeno L-12_H-512_A-8

tensorflow-tc-small-bert-bert-en-uncased-L-12-H-512-A-8

Link do TensorFlow Hub

BERT pequeno L-12_H-768_A-12

tensorflow-tc-small-bert-bert-en-uncased-L-12-H-768-A-12

Link do TensorFlow Hub

BERT grande uncased

tensorflow-tc-bert-en-uncased-L-24-H-1024-A-16-2

Link do TensorFlow Hub

BERT grande cased

tensorflow-tc-bert-en-cased-L-24-H-1024-A-16-2

Link do TensorFlow Hub

Máscara de palavras inteiras BERT grande uncased

tensorflow-tc-bert-en-wwm-uncased-L-24-H-1024-A-16-2

Link do TensorFlow Hub

Máscara de palavras inteiras BERT grande cased

tensorflow-tc-bert-en-wwm-cased-L-24-H-1024-A-16-2

Link do TensorFlow Hub

Base ALBERT

tensorflow-tc-albert-en-base

Link do TensorFlow Hub

ELECTRA Small++

tensorflow-tc-electra-small-1

Link do TensorFlow Hub

Base ELECTRA

tensorflow-tc-electra-base-1

Link do TensorFlow Hub

BERT Base Wikipédia e BooksCorpus

tensorflow-tc-experts-bert-wiki-books-1

Link do TensorFlow Hub

BERT Base Medline/PubMed

tensorflow-tc-experts-bert-pubmed-1

Link do TensorFlow Hub

Base Talking Heads

tensorflow-tc-talking-heads-base

Link do TensorFlow Hub

Base Talking Heads

tensorflow-tc-talking-heads-large

Link do TensorFlow Hub