Rond het snoeien af door het masker te verwijderen
Na het snoeien van een model moet je het afronden voordat je het opslaat of uitrolt. Dat betekent dat je het pruning-masker verwijdert, zodat het model niet langer bijhoudt welke gewichten op nul zijn gezet. Je gebruikt prune.remove() om deze stap te voltooien.
Deze oefening maakt deel uit van de cursus
Schaalbare AI-modellen met PyTorch Lightning
Oefeninstructies
- Rond het snoeien van de gewichtparameter af.
- Gebruik
prune.remove()om het masker te verwijderen en de definitieve gewichten op te slaan. - Print het model om te bevestigen dat het snoeien is verwijderd.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
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(____)