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
PyTorch Lightning ile Ölçeklenebilir AI Modelleri
kursunun bir parçasıdırEgzersiz 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ı interaktif egzersiz
Bu örnek kodu tamamlayarak bu egzersizi bitirin.
# 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)