Model referensi yang didukung
Cloud TPU menyediakan serangkaian model referensi yang dioptimalkan untuk pelatihan yang cepat dan akurat.
Cloud TPU mendukung rilis framework utama dan minor TensorFlow, PyTorch, dan JAX/FLAX berikut.
Penomoran rilis TensorFlow telah berubah dengan rilis 2.5.0. Nomor rilis utama TensorFlow diakhiri dengan '0' dan semua nomor 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 terbaru yang didukung, periksa apakah ada rilis patch untuk rilis utama. Jika demikian, Anda dapat menjalankan rilis patch terbaru yang didukung, bukan rilis utama.
| Framework | Versi utama | Kategori model | Model referensi | Versi yang didukung |
|---|---|---|---|---|
| PyTorch | 2.x | Klasifikasi gambar | ResNet-PyTorch | 1.13, 2.0 |
| JAX | terbaru | Model Bahasa Besar | MaxText | terbaru |
| JAX/FLAX | terbaru | Klasifikasi gambar | ResNet50 | terbaru |