Modelli di riferimento supportati
Cloud TPU fornisce un insieme di modelli di riferimento ottimizzati per un addestramento rapido e accurato.
Cloud TPU supporta le seguenti release di framework principali e secondarie di TensorFlow, PyTorch e JAX/FLAX.
La numerazione delle release di TensorFlow è cambiata con la release 2.5.0. I principali numeri di release di TensorFlow terminano con "0" e tutti i numeri di release di patch terminano con numeri maggiori di "0". Ad esempio, TF 2.10.0 è una release principale e TF 2.10.1 è una release secondaria.
Per eseguire la versione del framework più recente supportata, verifica se sono presenti release di patch per la release principale. In questo caso, puoi eseguire l'ultima patch supportata anziché la release principale.
Framework | Versione principale | Categoria del modello | Modelli di riferimento | Versioni supportate |
---|---|---|---|---|
TensorFlow | 2.x | Classificazione di immagini | ResNet-2.x, MNIST-2.x, EfficientNet-2.x | Vedi Versioni di TensorFlow supportate. |
Modellazione linguistica | Transformer-2.x, BERT-2.x | Vedi Versioni di TensorFlow supportate. | ||
Rilevamento di oggetti | RetinaNet-2.x | Vedi Versioni di TensorFlow supportate. | ||
Segmentazione dell'immagine | Mask-RCNN-2.x, ShapeMask-2.x | Vedi Versioni di TensorFlow supportate. | ||
Sistemi di consigli | DLRM-2.x, DCN-2.x, NCF-2.x | Vedi Versioni di TensorFlow supportate. | ||
PyTorch | 2.x | Classificazione di immagini | ResNet-PyTorch | 1,13; 2,0 |
Generazione di immagini | Diffusione stabile, | 1,13; 2,0 | ||
JAX | il più recente | linguistici di grandi dimensioni (LLM). | MaxText | più recente |
JAX/FLAX | il più recente | Classificazione di immagini | ResNet50 | più recente |