Doğrusal bir katmana budama uygula
Tam bağlı (doğrusal) bir katman kullanarak bir görüntü sınıflandırma modeli eğittik. Şimdi, model boyutunu performansı önemli ölçüde etkilemeden azaltmak için bu katmana yapılandırılmamış budama uygulamamız isteniyor. Bu görev için PyTorch'un l1_unstructured yöntemini kullanacağız.
Model model değişkeni olarak önceden yüklenmiştir.
Bu egzersiz, kursun bir parçasıdır
PyTorch Lightning ile Ölçeklenebilir AI Modelleri
Egzersiz talimatları
torch.nn.utilsiçinden budama modülünü içe aktar.- Son katman olan
model[3]üzerindel1_unstructuredbudamasını uygula ve ağırlıkların %30'unu buda.
Uygulamalı etkileşimli egzersiz
Bu egzersizi bu örnek kodu tamamlayarak deneyin.
# Import pruning module
import torch.nn.utils.prune as ____
# Before pruning
print(model)
# Apply L1 unstructured pruning to model[3]
prune.____(model[3], name="____", amount=____)
# After pruning
print(model)