Finalizza il pruning rimuovendo la maschera
Dopo aver applicato il pruning a un modello, devi finalizzarlo prima di salvarlo o distribuirlo. Questo significa rimuovere la maschera di pruning, così il modello non tiene più traccia dei pesi azzerati. Userai prune.remove() per completare questo passaggio.
Questo esercizio fa parte del corso
Modelli di AI scalabili con PyTorch Lightning
Istruzioni dell'esercizio
- Finalizza il pruning del parametro dei pesi.
- Usa
prune.remove()per rimuovere la maschera e salvare i pesi finali. - Stampa il modello per confermare la rimozione del pruning.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
import torch.nn.utils.prune as prune
# Before pruning
print(model)
# Finalize pruning by removing the pruning mask
prune.____(model[3], "____")
# Print model structure after pruning
print(____)