1. Aprende
    2. /
  2. Cursos
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

Ejercicio

Přidání nové konvoluční vrstvy

Vedoucí projektu ti připravil nový CNN model. Podíváme se na jeho architekturu a přidáme k němu novou konvoluční vrstvu.

Model je dostupný jako CNNModel. Balíčky torch a torch.nn jako nn jsou už naimportované.

Instrucciones

100 XP
  • Vytvoř instanci modelu ze třídy CNNModel a přistup ke konvolučním vrstvám.
  • Vytvoř novou konvoluční vrstvu, kde in_channels odpovídá hodnotě out_channels stávající vrstvy, out_channels je nastaveno na 32, stride i padding jsou nastaveny na 1 a kernel_size na 3; přiřaď ji do proměnné conv2.
  • Připoj novou vrstvu k modelu pod názvem "conv2".