Modelos de referência compatíveis
O Cloud TPU oferece um conjunto de modelos de referência otimizados para um treinamento rápido e preciso.
O Cloud TPU é compatível com as versões principais e secundárias a seguir do TensorFlow, do PyTorch e do JAX/FLAX.
A numeração das versões do TensorFlow foi alterada para a versão 2.5.0. Os números de versão principal do TensorFlow terminam com "0" e todos os números de versão de patch terminam com números maiores que "0". Por exemplo, o TF 2.10.0 é uma versão principal e o TF 2.10.1 é uma versão secundária.
Para executar a versão de framework com suporte mais recente, verifique se há versões de patch para a versão principal. Nesse caso, é possível executar a versão de patch com suporte mais recente em vez da versão principal.
| Framework | Versão principal | Categoria do modelo | Modelos de referência | Versões compatíveis |
|---|---|---|---|---|
| TensorFlow | 2.x | Classificação de imagens | ResNet-2.x, MNIST-2.x, EfficientNet-2.x | Consulte Versões compatíveis do TensorFlow. |
| Modelagem de linguagem | Transformer-2.x, BERT-2.x | Consulte Versões compatíveis do TensorFlow. | ||
| Detecção de objetos | RetinaNet-2.x | Consulte Versões compatíveis do TensorFlow. | ||
| Segmentação de imagens | Mask-RCNN-2.x, ShapeMask-2.x | Consulte Versões compatíveis do TensorFlow. | ||
| Sistemas de recomendação | DLRM-2.x, DCN-2.x, NCF-2.x | Consulte Versões compatíveis do TensorFlow. | ||
| PyTorch | 2.x | Classificação de imagens | ResNet-PyTorch | 1,13, 2,0 |
| Geração de imagens | Stable Diffusion, | 1,13, 2,0 | ||
| JAX (em inglês) | mais recente | Modelos de linguagem grandes | MaxText | mais recente |
| JAX/FLAX | mais recente | Classificação de imagens | ResNet50 | mais recente |