Se usó la API de Cloud Translation para traducir esta página.
Switch to English

Modelos de referencia compatibles

Cloud TPU proporciona el siguiente conjunto de modelos de referencia que están optimizados para un entrenamiento rápido y preciso.

Framework Versión principal Categoría del modelo Modelos de referencia Versiones compatibles
TensorFlow 2.x Image Classification ResNet-2.x,MNIST-2.x,EfficientNet-2.x, 2.1, 2.2, 2.3, 2.4
Modelado de lenguajes Transformer-2.x,BERT-2.x 2.1, 2.2, 2.3, 2.4
Modelado de lenguajes XlNet-2.x 2.4
Object Detection RetinaNet‑2.x 2.1, 2.2, 2.3, 2.4
Segmentación de imágenes Mask-RCNN-2.x 2.2, 2.3, 2.4
Segmentación de imágenes ShapeMask-2.x 2.3, 2.4
Sistemas de recomendación NCF-2.x 2.3, 2.4
1.x Image Classification ResNet, AmoebaNet, MNasNet, MNIST, EfficientNet 1.15
Modelado de lenguajes Transformer, BERT 1.15
Object Detection RetinaNet 1.15
PyTorch 1.x Image Classification ResNet-PyTorch 1.8.1
Modelado de lenguajes FairSeq Transformer,FairSeq RoBERTa 1.8.1
Sistemas de recomendación DLRM 1.8.1