Finalize a poda removendo a máscara.
Depois de podar um modelo, você precisa finalizá-lo antes de salvar ou implantar. Isso significa remover a máscara de poda para que o modelo não acompanhe mais quais pesos foram zerados. Você vai usar prune.remove() para concluir esta etapa.
Este exercício faz parte do curso
Modelos de IA escaláveis com PyTorch Lightning
Instruções do exercício
- Finalize a poda do parâmetro de peso.
- Use
prune.remove()para tirar a máscara e guardar os pesos finais. - Imprima o modelo para ver se a poda foi removida.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
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(____)