마스크 제거로 가지치기 최종 마무리
모델에 가지치기를 적용한 후 저장하거나 배포하기 전에 최종 마무리가 필요합니다. 즉, 어떤 가중치가 0으로 설정되었는지 추적하지 않도록 가지치기 마스크를 제거해야 합니다. 이 단계는 prune.remove()를 사용해 완료합니다.
이 연습은 강의의 일부입니다
PyTorch Lightning으로 만드는 확장 가능한 AI 모델
연습 안내
- 가중치 파라미터에 대한 가지치기를 최종 마무리하세요.
prune.remove()를 사용해 마스크를 제거하고 최종 가중치를 저장하세요.- 모델을 출력해 마스크 제거가 완료되었는지 확인하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
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(____)