Bagian ini berisi kumpulan tambahan halaman referensi API untuk proyek dan paket yang terpisah dari paket tensorflow
, tetapi tidak memiliki halaman subsitus khusus.
Repositori Model TensorFlow
Repositori TensorFlow Models menyediakan implementasi model state-of-the-art (SOTA).
Direktori resmi/proyek berisi kumpulan model SOTA yang menggunakan API tingkat tinggi TensorFlow. Mereka dimaksudkan untuk dipelihara dengan baik, diuji, dan selalu diperbarui dengan API TensorFlow terbaru.
Kode library yang digunakan untuk membangun dan melatih model ini tersedia sebagai paket pip. Anda dapat menginstalnya menggunakan:
$ pip install tensorflow-models-official # For the latest release
$ #or
$ pip install tf-models-nightly # For the nightly build
Untuk menginstal paket dari sumber, lihat petunjuk ini .
Paket pip tensorflow-models-official
berisi dua modul tingkat atas: tensorflow_models
dan orbit
. Anda dapat mengimpornya dengan:
import tensorflow_models as tfm
import orbit
Model Tensorflow
Modul tensorflow_models
menangani membangun model dan mengonfigurasi pelatihan. Fungsionalitas khusus aplikasi tersedia di bawah tfm.vision
dan tfm.nlp
.
Orbit
Modul orbit
mendefinisikan library yang fleksibel dan ringan untuk menulis kode loop pelatihan yang disesuaikan di TensorFlow. Orbit fleksibel tentang jenis model yang digunakannya. Anda dapat menggunakan Orbit untuk melatih Model Keras (sebagai alternatif dari Keras' Model.fit
), tetapi Anda tidak harus menggunakan Keras sama sekali. Orbit terintegrasi secara mulus dengan tf.distribute
dan mendukung berjalan di berbagai jenis perangkat (CPU, GPU, dan TPU).