Modelli TensorFlow Hub
I seguenti modelli preaddestrati sono disponibili per l'uso per il trasferimento dell'addestramento con l'algoritmo di classificazione del testo - TensorFlow.
I seguenti modelli variano in modo significativo in termini di dimensioni, numero di parametri del modello, tempo di addestramento e latenza di inferenza per un determinato set di dati. Il modello migliore per il tuo caso d'uso dipende dalla complessità del set di dati di ottimizzazione e dai requisiti richiesti in termini di tempo di addestramento, latenza di inferenza o precisione del modello.
| Nome modello | model_id |
Origine |
|---|---|---|
|
BERT Base Uncased |
|
|
|
BERT Base Cased |
|
|
|
BERT Base Multilingual Cased |
|
|
|
Small BERT L-2_H-128_A-2 |
|
|
|
Small BERT L-2_H-256_A-4 |
|
|
|
Small BERT L-2_H-512_A-8 |
|
|
|
Small BERT L-2_H-768_A-12 |
|
|
|
Small BERT L-4_H-128_A-2 |
|
|
|
Small BERT L-4_H-256_A-4 |
|
|
|
Small BERT L-4_H-512_A-8 |
|
|
|
Small BERT L-4_H-768_A-12 |
|
|
|
Small BERT L-6_H-128_A-2 |
|
|
|
Small BERT L-6_H-256_A-4 |
|
|
|
Small BERT L-6_H-512_A-8 |
|
|
|
Small BERT L-6_H-768_A-12 |
|
|
|
Small BERT L-8_H-128_A-2 |
|
|
|
Small BERT L-8_H-256_A-4 |
|
|
|
Small BERT L-8_H-512_A-8 |
|
|
|
Small BERT L-8_H-768_A-12 |
|
|
|
Small BERT L-10_H-128_A-2 |
|
|
|
Small BERT L-10_H-256_A-4 |
|
|
|
Small BERT L-10_H-512_A-8 |
|
|
|
Small BERT L-10_H-768_A-12 |
|
|
|
Small BERT L-12_H-128_A-2 |
|
|
|
Small BERT L-12_H-256_A-4 |
|
|
|
Small BERT L-12_H-512_A-8 |
|
|
|
Small BERT L-12_H-768_A-12 |
|
|
|
BERT Large Uncased |
|
|
|
BERT Large Cased |
|
|
|
BERT Large Uncased Whole Word Masking |
|
|
|
BERT Large Cased Whole Word Masking |
|
|
|
ALBERT Base |
|
|
|
ELECTRA Small++ |
|
|
|
ELECTRA Base |
|
|
|
BERT Base Wikipedia and BooksCorpus |
|
|
|
BERT Base MEDLINE/PubMed |
|
|
|
Talking Heads Base |
|
|
|
Talking Heads Large |
|
