Model referensi yang didukung
Cloud TPU menyediakan sekumpulan model referensi yang dioptimalkan untuk pelatihan yang cepat dan akurat.
Cloud TPU mendukung rilis framework utama dan minor berikut dari TensorFlow, PyTorch, dan JAX/FLAX.
Penomoran rilis TensorFlow telah berubah dengan rilis 2.5.0. Angka rilis TensorFlow utama diakhiri dengan '0' dan semua angka rilis patch diakhiri dengan angka yang lebih besar dari '0'. Misalnya, TF 2.10.0 adalah rilis utama dan TF 2.10.1 adalah rilis minor.
Untuk menjalankan versi framework yang didukung terbaru, periksa apakah ada rilis patch pada rilis utama. Jika demikian, Anda dapat menjalankan rilis patch yang didukung terbaru, bukan rilis utama.
| Framework | Versi utama | Kategori model | Model referensi | Versi yang didukung |
|---|---|---|---|---|
| TensorFlow | 2,x | Klasifikasi gambar | ResNet-2.x, MNIST-2.x, EfficientNet-2.x | Lihat Versi TensorFlow yang didukung. |
| Pemodelan bahasa | Transformer-2.x, BERT-2.x | Lihat Versi TensorFlow yang didukung. | ||
| Deteksi objek | RetinaNet-2.x | Lihat Versi TensorFlow yang didukung. | ||
| Segmentasi gambar | Mask-RCNN-2.x, ShapeMask-2.x | Lihat Versi TensorFlow yang didukung. | ||
| Sistem rekomendasi | DLRM-2.x, DCN-2.x, NCF-2.x | Lihat Versi TensorFlow yang didukung. | ||
| PyTorch | 2,x | Klasifikasi gambar | ResNet-PyTorch | 1,13, 2,0 |
| Pembuatan gambar | Difusi Stabil, | 1,13, 2,0 | ||
| JAX | terbaru | Model Bahasa Besar (LLM) | MaxText | terbaru |
| JAX/FLAX | terbaru | Klasifikasi gambar | ResNet50 | terbaru |