Questa pagina è stata tradotta dall'API Cloud Translation.
Switch to English

Ottimizza i modelli di machine learning

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(...)
TensorFlow Model Optimization Toolkit è una suite di strumenti per l'ottimizzazione dei modelli ML per la distribuzione e l'esecuzione. Tra i molti usi, il toolkit supporta le tecniche utilizzate per:
  • Riduci la latenza e i costi di inferenza per i dispositivi cloud ed edge (ad esempio, dispositivi mobili, IoT).
  • Distribuisci modelli su dispositivi periferici con limitazioni su elaborazione, memoria, consumo energetico, utilizzo della rete e spazio di archiviazione del modello.
  • Abilita l'esecuzione e ottimizza per hardware esistente o nuovi acceleratori per scopi speciali.

Scegli il modello e lo strumento di ottimizzazione a seconda della tua attività: