Maskeyi kaldırarak budamayı tamamla
Bir modeli budadıktan sonra, kaydetmeden veya dağıtıma almadan önce son haline getirmen gerekir. Bu, budama maskesini kaldırmak demektir; böylece model artık hangi ağırlıkların sıfırlandığını takip etmez. Bu adımı tamamlamak için prune.remove() kullanacaksın.
Bu egzersiz
PyTorch Lightning ile Ölçeklenebilir AI Modelleri
kursunun bir parçasıdırEgzersiz talimatları
- Ağırlık parametresinin budamasını tamamla.
- Maskeyi kaldırmak ve nihai ağırlıkları saklamak için
prune.remove()kullan. - Modeli yazdırarak maskenin kaldırıldığını doğrula.
Uygulamalı interaktif egzersiz
Bu örnek kodu tamamlayarak bu egzersizi bitirin.
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(____)