Modelos TensorFlow Hub - Amazon SageMaker AI

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

Modelos TensorFlow Hub

Tiene a su disposición los siguientes modelos prentrenados para usarlos en el aprendizaje por transferencia con el algoritmo Text Classification - TensorFlow.

El tamaño, la cantidad de parámetros, el tiempo de entrenamiento y la latencia de inferencia de estos modelos puede variar considerablemente para un conjunto de datos determinado. El mejor modelo para su caso de uso dependerá de la complejidad del conjunto de datos de ajuste y de sus necesidades con respecto al tiempo de entrenamiento, la latencia de inferencia o la precisión del modelo.

Nombre del modelo model_id Origen

BERT Base Uncased

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

Enlace a TensorFlow Hub

BERT Base Cased

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

Enlace a TensorFlow Hub

BERT Base Multilingual Cased

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

Enlace a TensorFlow Hub

Small BERT L-2_H-128_A-2

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

Enlace a TensorFlow Hub

Small BERT L-2_H-256_A-4

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

Enlace a TensorFlow Hub

Small BERT L-2_H-512_A-8

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

Enlace a TensorFlow Hub

Small BERT L-2_H-768_A-12

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

Enlace a TensorFlow Hub

Small BERT L-4_H-128_A-2

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

Enlace a TensorFlow Hub

Small BERT L-4_H-256_A-4

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

Enlace a TensorFlow Hub

Small BERT L-4_H-512_A-8

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

Enlace a TensorFlow Hub

Small BERT L-4_H-768_A-12

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

Enlace a TensorFlow Hub

Small BERT L-6_H-128_A-2

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

Enlace a TensorFlow Hub

Small BERT L-6_H-256_A-4

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

Enlace a TensorFlow Hub

Small BERT L-6_H-512_A-8

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

Enlace a TensorFlow Hub

Small BERT L-6_H-768_A-12

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

Enlace a TensorFlow Hub

Small BERT L-8_H-128_A-2

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

Enlace a TensorFlow Hub

Small BERT L-8_H-256_A-4

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

Enlace a TensorFlow Hub

Small BERT L-8_H-512_A-8

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

Enlace a TensorFlow Hub

Small BERT L-8_H-768_A-12

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

Enlace a TensorFlow Hub

Small BERT L-10_H-128_A-2

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

Enlace a TensorFlow Hub

Small BERT L-10_H-256_A-4

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

Enlace a TensorFlow Hub

Small BERT L-10_H-512_A-8

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

Enlace a TensorFlow Hub

Small BERT L-10_H-768_A-12

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

Enlace a TensorFlow Hub

Small BERT L-12_H-128_A-2

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

Enlace a TensorFlow Hub

Small BERT L-12_H-256_A-4

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

Enlace a TensorFlow Hub

Small BERT L-12_H-512_A-8

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

Enlace a TensorFlow Hub

Small BERT L-12_H-768_A-12

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

Enlace a TensorFlow Hub

BERT Large Uncased

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

Enlace a TensorFlow Hub

BERT Large Cased

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

Enlace a TensorFlow Hub

BERT Large Uncased Whole Word Masking

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

Enlace a TensorFlow Hub

BERT Large Cased Whole Word Masking

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

Enlace a TensorFlow Hub

ALBERT Base

tensorflow-tc-albert-en-base

Enlace a TensorFlow Hub

ELECTRA Small++

tensorflow-tc-electra-small-1

Enlace a TensorFlow Hub

ELECTRA Base

tensorflow-tc-electra-base-1

Enlace a TensorFlow Hub

BERT Base Wikipedia and BooksCorpus

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

Enlace a TensorFlow Hub

BERT Base MEDLINE/PubMed

tensorflow-tc-experts-bert-pubmed-1

Enlace a TensorFlow Hub

Talking Heads Base

tensorflow-tc-talking-heads-base

Enlace a TensorFlow Hub

Talking Heads Large

tensorflow-tc-talking-heads-large

Enlace a TensorFlow Hub