Modèles de référence compatibles
Cloud TPU fournit un ensemble de modèles de référence optimisés pour un entraînement rapide et précis.
Cloud TPU est compatible avec les versions de framework majeures et mineures suivantes de TensorFlow, PyTorch et JAX/FLAX.
La numérotation de version de TensorFlow a changé avec la version 2.5.0. Les numéros de versions principales de TensorFlow se terminent par "0", et tous les numéros de version de correctif se terminent par un chiffre supérieur à "0". Par exemple, TF 2.10.0 est une version majeure et TF 2.10.1 est une version mineure.
Afin d'exécuter la dernière version de framework compatible, vérifiez s'il existe des versions de correctif pour la version majeure. Dans ce cas, vous pouvez exécuter la dernière version de correctif compatible plutôt que la version majeure.
| Framework | Version majeure | Catégorie de modèle | Modèles de référence | Versions compatibles |
|---|---|---|---|---|
| TensorFlow | 2.x | Classification d'images | ResNet-2.x, MNIST-2.x, EffectiveNet-2.x | Consultez la page Versions compatibles de TensorFlow. |
| Modélisation du langage | Transformer-2.x, BERT-2.x | Consultez la page Versions compatibles de TensorFlow. | ||
| Détection d'objets | RetinaNet-2.x | Consultez la page Versions compatibles de TensorFlow. | ||
| Segmentation d'image | Mask-RCNN-2.x et ShapeMask-2.x | Consultez la page Versions compatibles de TensorFlow. | ||
| Systèmes de recommandation | DLRM-2.x, DCN-2.x, NCF-2.x | Consultez la page Versions compatibles de TensorFlow. | ||
| PyTorch | 2.x | Classification d'images | ResNet-PyTorch | 1.13, 2.0 |
| Génération d'images | Diffusion stable | 1.13, 2.0 | ||
| JAX | La plus récente | Grands modèles de langage | MaxText | latest |
| JAX/FLAX | La plus récente | Classification d'images | ResNet50 | latest |