Modelos de referencia compatibles
Cloud TPU proporciona un conjunto de modelos de referencia que están optimizados para un entrenamiento rápido y preciso.
Cloud TPU es compatible con las siguientes versiones principales y secundarias del framework de TensorFlow, PyTorch y JAX/FLAX.
La numeración de las versiones de TensorFlow cambió con la versión 2.5.0. Los números principales de las actualizaciones de TensorFlow terminan en “0” y todos los números de versiones de parches terminan en números superiores a “0”. Por ejemplo, TF 2.10.0 es una actualización principal y TF 2.10.1 es una versión secundaria.
Para ejecutar la última versión del framework compatible, verifica si hay versiones de parche en la versión principal. Si es así, puedes ejecutar la última versión de parche compatible en lugar de la versión principal.
| Framework | Versión principal | Categoría del modelo | Modelos de referencia | Versiones compatibles |
|---|---|---|---|---|
| TensorFlow | 2.x | Clasificación de imágenes | ResNet-2.x, MNIST-2.x y EfficientNet-2.x | Consulta Versiones de TensorFlow compatibles. |
| Modelado de lenguajes | Transformer-2.x, BERT-2.x | Consulta Versiones de TensorFlow compatibles. | ||
| Detección de objetos | RetinaNet‑2.x | Consulta Versiones de TensorFlow compatibles. | ||
| Segmentación de imágenes | Mask-RCNN-2.x, ShapeMask-2.x | Consulta Versiones de TensorFlow compatibles. | ||
| Sistemas de recomendación | DLRM-2.x, DCN-2.x, NCF-2.x | Consulta Versiones de TensorFlow compatibles. | ||
| PyTorch | 2.x | Clasificación de imágenes | ResNet-PyTorch | 1.13, 2.0 |
| Generación de imágenes | Dispersión estable. | 1.13, 2.0 | ||
| Jaime | Más reciente | Modelos grandes de lenguaje | MaxText | Más reciente |
| JAX/FLAX | Más reciente | Image Classification | ResNet50 | Más reciente |