Modelos oficiales compatibles

Cloud TPU con TensorFlow ofrece el siguiente conjunto de modelos oficiales y compatibles que se optimizaron para que el entrenamiento sea rápido y preciso.

Versión de TensorFlow Inicio de la compatibilidad Finalización de la compatibilidad Categoría del modelo Modelos compatibles
TF 2.3 28 de julio de 2020 (Fecha de finalización aún no establecida) Clasificación de imágenes ResNet-2.x,MNIST-2.x,EfficientNet-2.x,
Modelado de lenguajes Transformer-2.x,BERT-2.x
Detección de objetos RetinaNet‑2.x
Segmentación de imágenes Mask-RCNN-2.x, ShapeMask-2.x
Sistemas de recomendación NCF-2.x
TF 2.2 8 de mayo de 2020 (Fecha de finalización aún no establecida) Clasificación de imágenes ResNet-2.x,MNIST-2.x,EfficientNet-2.x,
Modelado de lenguajes Transformer-2.x,BERT-2.x
Detección de objetos RetinaNet‑2.x
Segmentación de imágenes Mask-RCNN-2.x
TF 2.1.1 21 de mayo de 2020 (Fecha de finalización aún no establecida) Clasificación de imágenes ResNet-2.x,MNIST-2.x
Modelado de lenguajes Transformer-2.x,BERT-2.x
Detección de objetos RetinaNet‑2.x
TF 2.1 9 de enero de 2020 30 de junio de 2020 Clasificación de imágenes ResNet-2.x,MNIST-2.x
Modelado de lenguajes Transformer-2.x,BERT-2.x
Detección de objetos RetinaNet‑2.x
TF 1.15.3 29 de mayo de 2020 (Fecha de finalización aún no establecida) Clasificación de imágenes ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Modelado de lenguajes Transformer, BERT
Detección de objetos RetinaNet
Segmentación de imágenes Mask RCNN, ShapeMask
TF 1.15.2 12 de mayo de 2020 30 de junio de 2020 Clasificación de imágenes ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Modelado de lenguajes Transformer, BERT
Detección de objetos RetinaNet
Segmentación de imágenes Mask RCNN, ShapeMask
TF 1.15 22 de octubre de 2019 30 de junio de 2020 Clasificación de imágenes ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
TF 1.14 19 de julio de 2019 30 de junio de 2020 Clasificación de imágenes ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Modelado de lenguajes
Detección de objetos RetinaNet, Mask R-CNN
TF 1.13 11 de marzo de 2019 30 de junio de 2020 Clasificación de imágenes ResNet, AmoebaNet,
MNasNet, MNIST
Modelado de lenguajes
Detección de objetos RetinaNet
Versión de PyTorch Inicio de la compatibilidad Finalización de la compatibilidad Categoría del modelo Modelos compatibles
PyTorch 1.x 17 de octubre de 2019 (Fecha de finalización aún no establecida) Clasificación de imágenes ResNet
Modelado de lenguajes FairSeq Transformer,FairSeq RoBERTa
Sistemas de recomendación DLRM