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.