1. Nauka
    2. /
  2. Kursy
    3. /
  3. Skalowalne modele AI z PyTorch Lightning
Connected

ćwiczenie

Finalizowanie przycinania przez usunięcie maski

Po przycięciu modelu należy go sfinalizować przed zapisaniem lub wdrożeniem. Oznacza to usunięcie maski przycinania, dzięki czemu model przestaje śledzić, które wagi zostały wyzerowane. Użyj prune.remove(), aby wykonać ten krok.

Instrukcje

100 XP
  • Sfinalizuj przycinanie parametru wag.
  • Użyj prune.remove(), aby usunąć maskę i zapisać końcowe wagi.
  • Wyświetl model, aby potwierdzić usunięcie maski przycinania.