Modèles officiels compatibles

Cloud TPU et TensorFlow fournissent l'ensemble suivant de modèles officiels compatibles qui ont été optimisés pour un entraînement rapide et précis.

Version de TensorFlow Date de début de compatibilité Date de fin de compatibilité Catégorie du modèle Modèles compatibles
tf 2.3 28 juillet 2020 (Date de fin pas encore fixée) Classification d'images ResNet-2.x, MNIST-2.x, EfficientNet-2.x,
Modélisation du langage Transformer-2.x, BERT-2.x
Détection d'objets RetinaNet-2.x
Segmentation d'image Mask-RCNN-2.x, ShapeMask-2.x
Systèmes de recommandation NCF-2.x
TF 2.2 8 mai 2020 (Date de fin pas encore fixée) Classification d'images ResNet-2.x, MNIST-2.x, EfficientNet-2.x,
Modélisation du langage Transformer-2.x, BERT-2.x
Détection d'objets RetinaNet-2.x
Segmentation d'image Mask-RCNN-2.x
TF 2.1.1 21 mai 2020 (Date de fin pas encore fixée) Classification d'images ResNet-2.x, MNIST-2.x
Modélisation du langage Transformer-2.x, BERT-2.x
Détection d'objets RetinaNet-2.x
TF 2.1 9 janvier 2020 30 juin 2020 Classification d'images ResNet-2.x, MNIST-2.x
Modélisation du langage Transformer-2.x, BERT-2.x
Détection d'objets RetinaNet-2.x
TF 1.15.3 29 mai 2020 (Date de fin pas encore fixée) Classification d'images ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Modélisation du langage Transformer,BERT
Détection d'objets RetinaNet
Segmentation d'image Mask R-CNN, ShapeMask
TF 1.15.2 12 mai 2020 30 juin 2020 Classification d'images ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Modélisation du langage Transformer,BERT
Détection d'objets RetinaNet
Segmentation d'image Mask R-CNN, ShapeMask
TF 1.15 22 octobre 2019 30 juin 2020 Classification d'images ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
TF 1.14 19 juillet 2019 30 juin 2020 Classification d'images ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Modélisation du langage
Détection d'objets RetinaNet, Mask R-CNN
TF 1.13 11 mars 2019 30 juin 2020 Classification d'images ResNet, AmoebaNet,
MNasNet, MNIST
Modélisation du langage
Détection d'objets RetinaNet
Version PyTorch Date de début de compatibilité Date de fin de compatibilité Catégorie du modèle Modèles compatibles
PyTorch 1.x 17 octobre 2019 (Date de fin pas encore fixée) Classification d'images ResNet
Modélisation du langage FairSeq Transformer, FairSeq RoBERTa
Systèmes de recommandation DLRM