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 admite las siguientes versiones principales y secundarias de los frameworks 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 actualización de TensorFlow terminan en “0” y todos los números de versión de parche terminan con números mayores que “0”. Por ejemplo, TF 2.10.0 es una versión principal y TF 2.10.1 es una versión secundaria.
Para ejecutar la versión de framework compatible más reciente, comprueba si hay versiones de parche para la versión principal. Si es así, puedes ejecutar la versión de parche compatible más reciente en lugar de la versión principal.
| Framework | Versión principal | Categoría del modelo | Modelos de referencia | Versiones compatibles |
|---|---|---|---|---|
| PyTorch | 2.x | Clasificación de imágenes | ResNet-PyTorch | 1.13 y 2.0 |
| JAX | Más reciente | Los modelos de lenguaje grandes | MaxText | Más reciente |
| JAX/FLAX | Más reciente | Image Classification | ResNet50 | Más reciente |