サポートされているリファレンス モデル
Cloud TPU には、高速かつ正確なトレーニング用に最適化された一連のリファレンス モデルが用意されています。
Cloud TPU は、TensorFlow、PyTorch、JAX/FLAX の以下のメジャーおよびマイナー フレームワークのリリースをサポートしています。
TensorFlow のリリース番号がリリース 2.5.0 で変更されました。メジャー TensorFlow リリース番号は「0」で終わり、すべてのパッチリリース番号は「0」より大きい数字で終わります。たとえば、TF 2.10.0 はメジャー リリースで、TF 2.10.1 はマイナー リリースです。
サポートされている最新バージョンのフレームワークを実行するには、メジャー リリースに対するパッチリリースがあるかどうかを確認します。その場合は、メジャー リリースではなく、サポートされている最新のパッチリリースを実行できます。
| フレームワーク | メジャー バージョン | モデルカテゴリ | リファレンス モデル | サポート対象のバージョン |
|---|---|---|---|---|
| TensorFlow | 2.x | 画像分類 | ResNet-2.x、MNIST-2.x、EfficientNet-2.x | サポートされている TensorFlow のバージョンをご覧ください。 |
| 言語モデリング | Transformer-2.x、BERT-2.x | サポートされている TensorFlow のバージョンをご覧ください。 | ||
| オブジェクト検出 | RetinaNet-2.x | サポートされている TensorFlow のバージョンをご覧ください。 | ||
| 画像セグメンテーション | Mask-RCNN-2.x、ShapeMask-2.x | サポートされている TensorFlow のバージョンをご覧ください。 | ||
| レコメンデーション システム | DLRM-2.x、 DCN-2.x、 NCF-2.x | サポートされている TensorFlow のバージョンをご覧ください。 | ||
| PyTorch | 2.x | 画像分類 | ResNet-PyTorch | 1.13, 2.0 |
| 画像の生成 | Stable Diffusion, | 1.13, 2.0 | ||
| JAX | 最新 | 大規模言語モデル | MaxText | 最新 |
| JAX / FLAX | 最新 | 画像分類 | ResNet50 | 最新 |