TensorFlow Modelos de hub - Amazon SageMaker

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 Origem

BERTBase sem estojo

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

TensorFlow Link do hub

BERTEstojo básico

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

TensorFlow Link do hub

BERTEstojo multilíngue básico

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

TensorFlow Link do hub

BERTL-2_H-128_A-2 pequeno

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

TensorFlow Link do hub

BERTL-2_H-256_A-4 pequeno

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

TensorFlow Link do hub

BERTL-2_H-512_A-8 pequeno

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

TensorFlow Link do hub

BERTL-2_H-768_A-12 pequeno

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

TensorFlow Link do hub

BERTL-4_H-128_A-2 pequeno

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

TensorFlow Link do hub

BERTL-4_H-256_A-4 pequeno

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

TensorFlow Link do hub

BERTL-4_H-512_A-8 pequeno

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

TensorFlow Link do hub

BERTL-4_H-768_A-12 pequeno

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

TensorFlow Link do hub

BERTL-6_H-128_A-2 pequeno

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

TensorFlow Link do hub

BERTL-6_H-256_A-4 pequeno

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

TensorFlow Link do hub

BERTL-6_H-512_A-8 pequeno

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

TensorFlow Link do hub

BERTL-6_H-768_A-12 pequeno

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

TensorFlow Link do hub

BERTL-8_H-128_A-2 pequeno

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

TensorFlow Link do hub

BERTL-8_H-256_A-4 pequeno

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

TensorFlow Link do hub

BERTL-8_H-512_A-8 pequeno

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

TensorFlow Link do hub

BERTL-8_H-768_A-12 pequeno

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

TensorFlow Link do hub

BERTL-10_H-128_A-2 pequeno

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

TensorFlow Link do hub

BERTL-10_H-256_A-4 pequeno

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

TensorFlow Link do hub

BERTL-10_H-512_A-8 pequeno

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

TensorFlow Link do hub

BERTL-10_H-768_A-12 pequeno

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

TensorFlow Link do hub

BERTL-12_H-128_A-2 pequeno

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

TensorFlow Link do hub

BERTL-12_H-256_A-4 pequeno

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

TensorFlow Link do hub

BERTL-12_H-512_A-8 pequeno

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

TensorFlow Link do hub

BERTL-12_H-768_A-12 pequeno

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

TensorFlow Link do hub

BERTGrande, sem caixa

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

TensorFlow Link do hub

BERTEstojo grande

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

TensorFlow Link do hub

BERTMascaramento de palavras inteiras grande e sem caixa

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

TensorFlow Link do hub

BERTMascaramento de palavras inteiras em caixa grande

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

TensorFlow Link do hub

ALBERTBase

tensorflow-tc-albert-en-base

TensorFlow Link do hub

ELECTRAPequeno++

tensorflow-tc-electra-small-1

TensorFlow Link do hub

ELECTRABase

tensorflow-tc-electra-base-1

TensorFlow Link do hub

BERTWikipédia base e BooksCorpus

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

TensorFlow Link do hub

BERTBaseMEDLINE/PubMed

tensorflow-tc-experts-bert-pubmed-1

TensorFlow Link do hub

Base Talking Heads

tensorflow-tc-talking-heads-base

TensorFlow Link do hub

Base Talking Heads

tensorflow-tc-talking-heads-large

TensorFlow Link do hub