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 principali e secondarie dei framework TensorFlow, PyTorch e JAX/FLAX.
La numerazione delle release di TensorFlow è cambiata con la release 2.5.0. I numeri delle release principali di TensorFlow terminano con "0" e tutti i numeri delle release delle patch terminano con numeri superiori a "0". Ad esempio, TF 2.10.0 è una release principale e TF 2.10.1 è una release secondaria.
Per eseguire la versione più recente del framework supportata, controlla se sono presenti release di patch per la release principale. In questo caso, puoi eseguire la release di patch supportata più recente anziché la release principale.
| Framework | Versione principale | Categoria del modello | Modelli di riferimento | Versioni supportate |
|---|---|---|---|---|
| PyTorch | 2.x | Classificazione delle immagini | ResNet-PyTorch | 1.13, 2.0 |
| JAX | ultime | Modelli linguistici di grandi dimensioni (LLM) | MaxText | più recenti |
| JAX/FLAX | ultime | Classificazione delle immagini | ResNet50 | più recenti |