Een nieuwe convolutionele laag toevoegen

Je projectleider heeft je een nieuw CNN-model gegeven. Laten we de architectuur van het model bekijken en er een nieuwe convolutionele laag aan toevoegen.

Het model is beschikbaar als CNNModel. De pakketten torch en torch.nn als nn zijn geïmporteerd.

Deze oefening maakt deel uit van de cursus

Deep Learning voor afbeeldingen met PyTorch

Cursus bekijken

Oefeninstructies

Maak een instantie van het model uit de klasse CNNModel en bekijk de convolutionele lagen.
Maak een nieuwe convolutionele laag met in_channels gelijk aan de out_channels van de bestaande laag, out_channels ingesteld op 32, stride en padding beide op 1, en een kernel_size van 3; wijs deze toe aan conv2.
Voeg de nieuwe laag toe aan het model en noem hem "conv2".

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

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

Code bewerken en uitvoeren