官方支持的模型

带 TensorFlow 的 Cloud TPU 提供下列官方支持的模型,这些模型已经过优化,可实现快速准确的训练。

TensorFlow 版本 支持开始日期 支持结束日期 模型类别 支持的模型
TF 2.3 2020 年 7 月 28 日 (结束日期尚未确定) 图片分类 ResNet-2.xMNIST-2.xEfficientNet-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
图片分割 Mask-RCNN-2.xShapeMask-2.x
推荐系统 NCF-2.x
TF 2.2 2020 年 5 月 8 日 (结束日期尚未确定) 图片分类 ResNet-2.xMNIST-2.xEfficientNet-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
图片分割 Mask-RCNN-2.x
TF 2.1.1 2020 年 5 月 21 日 (结束日期尚未确定) 图片分类 ResNet-2.xMNIST-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
TF 2.1 2020 年 1 月 9 日 2020 年 6 月 30 日 图片分类 ResNet-2.xMNIST-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
TF 1.15.3 2020 年 5 月 29 日 (结束日期尚未确定) 图片分类 ResNetAmoebaNet
EfficientNet
MNasNetMNIST
语言建模 TransformerBERT
对象检测 RetinaNet
图片分割 Mask RCNNShapeMask
TF 1.15.2 2020 年 5 月 12 日 2020 年 6 月 30 日 图片分类 ResNetAmoebaNet
EfficientNet
MNasNetMNIST
语言建模 TransformerBERT
对象检测 RetinaNet
图片分割 Mask RCNNShapeMask
TF 1.15 2019 年 10 月 22 日 2020 年 6 月 30 日 图片分类 ResNetAmoebaNet
EfficientNet
MNasNetMNIST
TF 1.14 2019 年 7 月 19 日 2020 年 6 月 30 日 图片分类 ResNetAmoebaNet
EfficientNet
MNasNetMNIST
语言建模
对象检测 RetinaNetMask RCNN
TF 1.13 2019 年 3 月 11 日 2020 年 6 月 30 日 图片分类 ResNetAmoebaNet
MNasNetMNIST
语言建模
对象检测 RetinaNet
PyTorch 版本 支持开始日期 支持结束日期 模型类别 支持的模型
PyTorch 1.x 2016 年 10 月 27 日 (结束日期尚未确定) 图片分类 ResNet
语言建模 FairSeq TransformerFairSeq RoBERTa
推荐系统 DLRM