Cloud TPU with TensorFlow provides the following set of official, supported models that have been optimized for fast and accurate training.
TensorFlow version | Support start | Support end | Model category | Supported models |
---|---|---|---|---|
TF 2.4.0 | December 21, 2020 | (End date not yet set) | Image classification | ResNet-2.x, MNIST-2.x, EfficientNet-2.x, |
Language modeling | Transformer-2.x, BERT-2.x | |||
Object detection | RetinaNet-2.x | |||
Image segmentation | Mask-RCNN-2.x, ShapeMask-2.x | |||
Recommendation systems | NCF-2.x | |||
TF 2.3.1 | October 1, 2020 | (End date not yet set) | Image classification | ResNet-2.x, MNIST-2.x, EfficientNet-2.x, |
Language modeling | Transformer-2.x, BERT-2.x | |||
Object detection | RetinaNet-2.x | |||
Image segmentation | Mask-RCNN-2.x, ShapeMask-2.x | |||
Recommendation systems | NCF-2.x | |||
TF 2.3 | July 28, 2020 | (End date not yet set) | Image classification | ResNet-2.x, MNIST-2.x, EfficientNet-2.x, |
Language modeling | Transformer-2.x, BERT-2.x | |||
Object detection | RetinaNet-2.x | |||
Image segmentation | Mask-RCNN-2.x, ShapeMask-2.x | |||
Recommendation systems | NCF-2.x | |||
TF 2.2.2 | January 12, 2020 | (End date not yet set) | Image classification | ResNet-2.x, MNIST-2.x, EfficientNet-2.x, |
Language modeling | Transformer-2.x, BERT-2.x | |||
Object detection | RetinaNet-2.x | |||
Image segmentation | Mask-RCNN-2.x | |||
TF 2.2.1 | October 7, 2020 | (End date not yet set) | Image classification | ResNet-2.x, MNIST-2.x, EfficientNet-2.x, |
Language modeling | Transformer-2.x, BERT-2.x | |||
Object detection | RetinaNet-2.x | |||
Image segmentation | Mask-RCNN-2.x | |||
TF 2.2 | May 8, 2020 | (End date not yet set) | Image classification | ResNet-2.x, MNIST-2.x, EfficientNet-2.x, |
Language modeling | Transformer-2.x, BERT-2.x | |||
Object detection | RetinaNet-2.x | |||
Image segmentation | Mask-RCNN-2.x | |||
TF 2.1.2 | October 30, 2020 | (End date not yet set) | Image classification | ResNet-2.x, MNIST-2.x |
Language modeling | Transformer-2.x, BERT-2.x | |||
Object detection | RetinaNet-2.x | |||
TF 2.1.1 | May 21, 2020 | (End date not yet set) | Image classification | ResNet-2.x, MNIST-2.x |
Language modeling | Transformer-2.x, BERT-2.x | |||
Object detection | RetinaNet-2.x | |||
TF 2.1 | January 9, 2020 | June 30, 2020 | Image classification | ResNet-2.x, MNIST-2.x |
Language modeling | Transformer-2.x, BERT-2.x | |||
Object detection | RetinaNet-2.x | |||
TF 1.15.5 | January 14, 2020 | (End date not yet set) | Image classification |
ResNet,
AmoebaNet, EfficientNet, MNasNet, MNIST |
Language modeling | Transformer, BERT | |||
Object detection | RetinaNet | |||
Image segmentation | Mask RCNN, ShapeMask | |||
TF 1.15.4 | October 21, 2020 | (End date not yet set) | Image classification |
ResNet,
AmoebaNet, EfficientNet, MNasNet, MNIST |
Language modeling | Transformer, BERT | |||
Object detection | RetinaNet | |||
Image segmentation | Mask RCNN, ShapeMask | |||
TF 1.15.3 | May 29, 2020 | (End date not yet set) | Image classification |
ResNet,
AmoebaNet, EfficientNet, MNasNet, MNIST |
Language modeling | Transformer, BERT | |||
Object detection | RetinaNet | |||
Image segmentation | Mask RCNN, ShapeMask | |||
TF 1.15.2 | May 12, 2020 | June 30, 2020 | Image classification |
ResNet,
AmoebaNet, EfficientNet, MNasNet, MNIST |
Language modeling | Transformer, BERT | |||
Object detection | RetinaNet | |||
Image segmentation | Mask RCNN, ShapeMask | |||
TF 1.15 | October 22, 2019 | June 30, 2020 | Image classification |
ResNet,
AmoebaNet, EfficientNet, MNasNet, MNIST |
TF 1.14 | July 19, 2019 | June 30, 2020 | Image classification |
ResNet,
AmoebaNet, EfficientNet, MNasNet, MNIST |
Language modeling | ||||
Object detection | RetinaNet, Mask RCNN | |||
TF 1.13 | March 11, 2019 | June 30, 2020 | Image classification |
ResNet,
AmoebaNet, MNasNet, MNIST |
Language modeling | ||||
Object detection | RetinaNet |
PyTorch version | Support start | Support end | Model category | Supported models |
---|---|---|---|---|
PyTorch 1.x | Oct 17, 2019 | (End date not yet set) | Image classification | ResNet |
Language modeling | FairSeq Transformer, FairSeq RoBERTa | |||
Recommendation systems | DLRM |