1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Introduction to Deep Learning with PyTorch
Connected

cvičení

Zmrazení vrstev modelu

Chystáš se doladit model na nový úkol po načtení předtrénovaných vah. Model obsahuje tři lineární vrstvy. Protože je ale tvůj dataset malý, chceš trénovat pouze poslední lineární vrstvu a první dvě lineární vrstvy zmrazit.

Model už byl vytvořen a je uložen v proměnné model. K zobrazení parametrů modelu použiješ metodu named_parameters. Každý parametr je popsán názvem – jde o řetězec ve formátu x.name, kde x je index dané vrstvy.

Měj na paměti, že lineární vrstva má dva parametry: weight (váhy) a bias.

Pokyny

100 XP
  • Pomocí příkazu if zkontroluj, zda parametr patří mezi váhy první nebo druhé vrstvy.
  • Zmraz váhy prvních dvou vrstev tohoto modelu.