IniziaInizia gratis

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

Visualizza il corso

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(____)
Modifica ed esegui il codice