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é.

Toto cvičení je součástí kurzu

Deep Learning pro obrázky s PyTorchem

Zobrazit kurz

Pokyny k cvičení

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".

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# Create a model
model = ____
print("Original model: ", model)

# Create a new convolutional layer
conv2 = ____

# Append the new layer to the model
model.____(____)
print("Extended model: ", model)

Upravit a spustit kód