Modelos compatíveis

O Cloud TPU fornece o seguinte conjunto de modelos compatíveis que são otimizados para treinamento rápido e preciso.

Versão do TensorFlow Data de início do suporte Data de conclusão do suporte Categoria do modelo Modelos compatíveis
TF 2.3 28 de julho de 2020 (data final ainda não definida) Classificação de imagens ResNet-2.x, MNIST-2.x, EfficientNet-2.x,
Modelagem de linguagem Transformer-2.x, BERT-2.x
Detecção de objetos RetinaNet-2.x
Segmentação de imagens Mask-RCNN-2.x, ShapeMask-2.x
Sistemas de recomendação NCF-2.x
TF 2.2 8 de maio de 2020 (data final ainda não definida) Classificação de imagens ResNet-2.x, MNIST-2.x, EfficientNet-2.x,
Modelagem de linguagem Transformer-2.x, BERT-2.x
Detecção de objetos RetinaNet-2.x
Segmentação de imagens Mask-RCNN-2.x
TF 2.1.1 21 de maio de 2020 (data final ainda não definida) Classificação de imagens ResNet-2.x, MNIST-2.x
Modelagem de linguagem Transformer-2.x, BERT-2.x
Detecção de objetos RetinaNet-2.x
TF 2.1 9 de janeiro de 2020 30 de junho de 2020 Classificação de imagens ResNet-2.x, MNIST-2.x
Modelagem de linguagem Transformer-2.x, BERT-2.x
Detecção de objetos RetinaNet-2.x
TF 1.15.3 29 de maio de 2020 (data final ainda não definida) Classificação de imagens ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Modelagem de linguagem Transformer, BERT
Detecção de objetos RetinaNet
Segmentação de imagens Mask RCNN, ShapeMask
TF 1.15.2 12 de maio de 2020 30 de junho de 2020 Classificação de imagens ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Modelagem de linguagem Transformer, BERT
Detecção de objetos RetinaNet
Segmentação de imagens Mask RCNN, ShapeMask
TF 1.15 22 de outubro de 2019 30 de junho de 2020 Classificação de imagens ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
TF 1.14 19 de julho de 2019 30 de junho de 2020 Classificação de imagens ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Modelagem de linguagem
Detecção de objetos RetinaNet, Mask RCNN
TF 1.13 11 de março de 2019 30 de junho de 2020 Classificação de imagens ResNet, AmoebaNet,
MNasNet, MNIST
Modelagem de linguagem
Detecção de objetos RetinaNet
Versão PyTorch Data de início do suporte Data de conclusão do suporte Categoria do modelo Modelos compatíveis
PyTorch 1.x 17 de outubro de 2019 (data final ainda não definida) Classificação de imagens ResNet
Modelagem de linguagem FairSeq Transformer, FairSeq RoBERTa
Sistemas de recomendação DLRM