支持的参考模型

Cloud TPU 提供以下一组参考模型,这些模型经过优化,可实现快速准确的训练。

框架 主要版本 模型类别 参考模型 支持的版本
TensorFlow 2.x 图片分类 ResNet-2.xMNIST-2.xEfficientNet-2.x 2.1、2.2、2.3、2.4
语言建模 Transformer-2.xBERT-2.x 2.1、2.2、2.3、2.4
语言建模 XlNet-2.x 2.4
对象检测 RetinaNet-2.x 2.1、2.2、2.3、2.4
图片分割 Mask-RCNN-2.x 2.2、2.3、2.4
图片分割 ShapeMask-2.x 2.3、2.4
推荐系统 NCF-2.x 2.3、2.4
1.x 图片分类 ResNetAmoebaNetMNasNetMNISTEfficientNet 1.15
语言建模 TransformerBERT 1.15
对象检测 RetinaNet 1.15
PyTorch 1.x 图片分类 ResNet-PyTorch 1.7
语言建模 FairSeq TransformerFairSeq RoBERTa 1.7
推荐系统 DLRM 1.7