Image Classification
-
Running MNIST on Cloud TPU (TF 2.x)
An MNIST image classification model using TensorFlow, optimized to run on Cloud TPU.
-
Running MNIST on Cloud TPU (TF 1.x)
An MNIST image classification model using TensorFlow, optimized to run on Cloud TPU.
-
Training ResNet on Cloud TPU (TF 2.x)
A ResNet image classification model using TensorFlow, optimized to run on Cloud TPU.
-
Training Keras ResNet-RS on Cloud TPU (TF 2.x)
A Keras ResNet-RS model using TensorFlow, optimized to run on Cloud TPU.
-
Training ResNet on Cloud TPU (TF 1.x)
A ResNet image classification model using TensorFlow, optimized to run on Cloud TPU.
-
(PyTorch) Training ResNet on Cloud TPU
A ResNet image classification model using PyTorch, optimized to run on Cloud TPU.
-
Training MnasNet on Cloud TPU (TF 1.x)
An MnasNet image classification model using TensorFlow, optimized to run on Cloud TPU.
-
Training EfficientNet on Cloud TPU (TF 2.x)
An EfficientNet image classification model using TensorFlow, optimized to run on Cloud TPU.
-
Training EfficientNet on Cloud TPU (TF 1.x)
An EfficientNet image classification model using TensorFlow, optimized to run on Cloud TPU.
-
Training AmoebaNet-D on Cloud TPU (TF 1.x)
An AmoebaNet image classification model using TensorFlow, optimized to run on Cloud TPU.
-
Running Inception on Cloud TPU (TF 1.x)
A guide to training the Inception models on Cloud TPU.
-
Running Deeplab v3 on Cloud TPU (TF 1.x)
A guide to training the Deeplab v3 model on Cloud TPU.
Object Detection
-
Training RetinaNet on Cloud TPU (TF 1.x)
A RetinaNet object detection model using TensorFlow, optimized to run on Cloud TPU.
-
Training RetinaNet on Cloud TPU (TF 2.x)
A RetinaNet object detection model using TensorFlow, optimized to run on Cloud TPU.
-
Training ShapeMask on Cloud TPU (TF 2.x)
A ShapeMask object detection model using TensorFlow, optimized to run on Cloud TPU.
-
Training ShapeMask on Cloud TPU (TF 1.x)
A ShapeMask object detection model using TensorFlow, optimized to run on Cloud TPU.
-
Training Mask RCNN on Cloud TPU (TF 2.x)
A Mask RCNN model using TensorFlow, optimized to run on Cloud TPU.
-
Training Mask RCNN on Cloud TPU (TF 1.x)
A Mask RCNN model using TensorFlow, optimized to run on Cloud TPU.
Recommendation Systems
-
(PyTorch) Training DLRM model on Cloud TPU
A guide to training the PyTorch DLRM model on Cloud TPU.
-
Training DLRM and DCN models on Cloud TPU (TF 2.x)
A guide to training DLRM and DCN v2 ranking models for tasks such as click-through rate (CTR) prediction.
-
Training a Neural Collaboration Filtering model on Cloud TPU (TF 2.x)
An implementation of the Neural Collaborative Filtering (NCF) framework with the Neural Matrix Factorization (NeuMF) model
Distributed Processing on a Pod
Natural Language Processing
-
BERT FineTuning with Cloud TPU: Sentence and Sentence-Pair Classification Tasks (TF 2.x)
Discover how to use Bidirectional Encoder Representations from Transformers (BERT) with Cloud TPU.
-
BERT FineTuning with Cloud TPU: Sentence and Sentence-Pair Classification Tasks (TF 1.x)
Discover how to use Bidirectional Encoder Representations from Transformers (BERT) with Cloud TPU.
-
(PyTorch) Pre-training FairSeq RoBERTa on Cloud TPU
A guide to pre-training the FairSeq version of the RoBERTa model on Cloud TPU using the public wikitext-103 dataset.
T2T Transformer
-
Training Transformer on Cloud TPU (TF 2.x)
A guide to training the Tensor2Tensor Transformer model on Cloud TPU, for English-German translation.
-
Running the Transformer with Tensor2Tensor on Cloud TPU (TF 1.x)
A guide to training the Tensor2Tensor Transformer model on Cloud TPU, for translation, language modeling, and sentiment analysis.
-
(PyTorch) Training FairSeq Transformer on Cloud TPU
A guide to training the FairSeq version of the Transformer model on Cloud TPU and running the WMT 18 translation task translating English to German.
-
Running Automatic Speech Recognition on Cloud TPU (TF 1.x)
A voice-text model using TensorFlow, optimized to run on Cloud TPU.