教程

图片分类

• 在 Cloud TPU 上运行 MNIST (TF 2.x)

  一个使用 TensorFlow 的 MNIST 图片分类模型，针对在 Cloud TPU 上运行进行了优化。

• 在 Cloud TPU 上训练 ResNet (TF 2.x)

  一个使用 TensorFlow 的 ResNet 图片分类模型，针对在 Cloud TPU 上运行进行了优化。

• 在 Cloud TPU (TF 2.x) 上训练 Keras ResNet-RS

  一个使用 TensorFlow 的 Keras ResNet-RS 模型，针对在 Cloud TPU 上运行进行了优化。

• 在 Cloud TPU 上训练 ResNet (PyTorch)

  一个使用 PyTorch 的 ResNet 图片分类模型，针对在 Cloud TPU 上运行进行了优化。

• 在 Cloud TPU 上训练 EfficientNet (TF 2.x)

  一个使用 TensorFlow 的 EfficientNet 图片分类模型，针对在 Cloud TPU 上运行进行了优化。

对象检测

• 在 Cloud TPU 上训练 RetinaNet (TF 2.x)

  一个使用 TensorFlow 的 RetinaNet 对象检测模型，针对在 Cloud TPU 上运行进行了优化。

• 在 Cloud TPU 上训练 ShapeMask (TF 2.x)

  一个使用 TensorFlow 的 ShapeMask 对象检测模型，针对在 Cloud TPU 上运行进行了优化。

• 在 Cloud TPU 上训练 Mask RCNN (TF 2.x)

  一个使用 TensorFlow 的 Mask RCNN 模型，针对在 Cloud TPU 上运行进行了优化。

推荐系统

• 在 Cloud TPU (PyTorch) 上训练 DLRM 模型

  在 Cloud TPU 上训练 PyTorch DLRM 模型的指南。

• 在 Cloud TPU 上训练 DLRM 和 DCN 模型 (TF 2.x)

  为点击率 (CTR) 预测等任务训练 DLRM 和 DCN v2 排名模型的指南。

• 在 Cloud TPU 上训练神经协同过滤模型 (TF 2.x)

  实现了采用神经矩阵分解 (NeuMF) 模型的神经协同过滤 (NCF) 框架

Pod 上的分布式处理

• 在 Cloud TPU Pod (PyTorch) 上训练模型

  一个使用 PyTorch 的 ResNet-50 图片分类模型，针对在 Cloud TPU Pod 上运行进行了优化。

自然语言处理

• 使用 Cloud TPU 调整 BERT 模型：句子和句对分类任务 (TF 2.x)

  了解如何将 Transformer 的双向编码器表示法 (BERT) 与 Cloud TPU 搭配使用。

• 在 Cloud TPU (PyTorch) 上预训练 FairSeq RoBERTa

  使用公开 Wikitext-103 数据集在 Cloud TPU 上预训练 FairSeq 版 RoBERTa 模型的指南。

大型语言模型

• 使用 Pax 在单设备 TPU 上训练

  在单设备 Cloud TPU 上使用 Pax 训练 SPMD 模型的指南。

• 使用 Pax 在 TPU Pod 切片上训练

  在 TPU Pod 上使用 Pax 训练 SPMD 模型的指南。