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