Schließ das Beschneiden ab, indem du die Maske entfernst.
Nachdem du ein Modell bereinigt hast, musst du es fertigstellen, bevor du es speichern oder bereitstellen kannst. Das heißt, die Beschneidungsmaske wird entfernt, sodass das Modell nicht mehr nachverfolgt, welche Gewichte auf Null gesetzt wurden. Du brauchst „ prune.remove()
“, um diesen Schritt zu erledigen.
Diese Übung ist Teil des Kurses
Skalierbare KI-Modelle mit PyTorch Lightning
Anleitung zur Übung
- Schließ das Beschneiden des Gewichtsparameters ab.
- Mit „
prune.remove()
“ kannst du die Maske entfernen und die endgültigen Gewichte speichern. - Druck das Modell aus, um zu checken, ob alles richtig gekürzt wurde.
Interaktive Übung
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
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(____)