1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch Lightning으로 만드는 확장 가능한 AI 모델
Connected

연습 문제

마스크 제거로 가지치기 최종 마무리

모델에 가지치기를 적용한 후 저장하거나 배포하기 전에 최종 마무리가 필요합니다. 즉, 어떤 가중치가 0으로 설정되었는지 추적하지 않도록 가지치기 마스크를 제거해야 합니다. 이 단계는 prune.remove()를 사용해 완료합니다.

지침

100 XP
  • 가중치 파라미터에 대한 가지치기를 최종 마무리하세요.
  • prune.remove()를 사용해 마스크를 제거하고 최종 가중치를 저장하세요.
  • 모델을 출력해 마스크 제거가 완료되었는지 확인하세요.