1. Learn
  2. /
  3. 课程
  4. /
  5. Skalowalne modele AI z PyTorch Lightning

Connected

道练习

Zastosowanie przycinania do warstwy liniowej

Wytrenowaliśmy model klasyfikacji obrazów z w pełni połączoną warstwą liniową. Teraz zastosujemy do tej warstwy przycinanie niestrukturalne (unstructured pruning), aby zmniejszyć rozmiar modelu bez istotnego wpływu na jego wydajność. Skorzystamy z metody l1_unstructured dostępnej w PyTorch.

Model jest wstępnie załadowany jako zmienna model.

说明

100 XP
  • Zaimportuj moduł przycinania z torch.nn.utils.
  • Zastosuj przycinanie l1_unstructured do model[3] – ostatniej warstwy – i przytnij 30% wag.