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à:
-
Migliora le prestazioni con i modelli standard
In molti casi, i modelli pre-ottimizzati possono migliorare l'efficienza dell'applicazione. -
Utilizza il Toolkit di ottimizzazione del modello TensorFlow
Prova gli strumenti di post-formazione per ottimizzare un modello TensorFlow già addestrato. -
Ottimizza ulteriormente
Usa strumenti di ottimizzazione del tempo di formazione e impara a conoscere le tecniche.