Halaman ini diterjemahkan oleh Cloud Translation API.
Switch to English

Mengoptimalkan model pembelajaran mesin

import tensorflow as tf
import tensorflow_model_optimization as tfmot

model = tf.keras.Sequential([...])

pruning_schedule = tfmot.sparsity.keras.PolynomialDecay(
                      initial_sparsity=0.0, final_sparsity=0.5,
                      begin_step=2000, end_step=4000)

model_for_pruning = tfmot.sparsity.keras.prune_low_magnitude(
    model, pruning_schedule=pruning_schedule)
...

model_for_pruning.fit(...)
Toolkit Pengoptimalan Model TensorFlow adalah seperangkat alat untuk mengoptimalkan model ML untuk penerapan dan eksekusi. Di antara banyak kegunaan, toolkit mendukung teknik yang digunakan untuk:
  • Kurangi biaya latensi dan inferensi untuk perangkat cloud dan edge (mis. Seluler, IoT).
  • Menerapkan model ke perangkat edge dengan batasan pada pemrosesan, memori, konsumsi daya, penggunaan jaringan, dan ruang penyimpanan model.
  • Aktifkan eksekusi dan optimalkan perangkat keras yang ada atau akselerator tujuan khusus baru.

Pilih model dan alat pengoptimalan tergantung pada tugas Anda: