Lägga till ett nytt konvolutionslager

Din projektledare har gett dig en ny CNN-modell. Nu ska vi titta på modellens arkitektur och lägga till ett nytt konvolutionslager.

Modellen finns tillgänglig som CNNModel. Paketen torch och torch.nn som nn har redan importerats.

Den här övningen är en del av kursen

Djupinlärning för bilder med PyTorch

Visa kurs

Övningsinstruktioner

Skapa en instans av modellen från klassen CNNModel och hämta konvolutionslagren.
Skapa ett nytt konvolutionslager där in_channels är lika med det befintliga lagrets out_channels, out_channels satt till 32, stride och padding båda satta till 1, och kernel_size satt till 3; tilldela det till conv2.
Lägg till det nya lagret i modellen och ge det namnet "conv2".

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# 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)

Redigera och kör kod