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 exercicio faz parte do curso
Modelos de IA escaláveis com PyTorch Lightning
Instruções do exercicio
- 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.
exercicio interativo prático
Tente este exercicio 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(____)