Eine neue Convolutional-Schicht hinzufügen

Dein Projektlead hat dir ein neues CNN-Modell bereitgestellt. Lass uns einen Blick auf die Architektur werfen und eine neue Convolutional-Schicht anhängen.

Das Modell ist als CNNModel verfügbar. Die Pakete torch und torch.nn als nn wurden importiert.

Diese Übung ist Teil des Kurses

<Kurs>Deep Learning für Bilder mit PyTorch</Kurs>

Kurs ansehen

Übungsanweisungen

Instanziiere ein Modell aus der Klasse CNNModel und greife auf die Convolutional-Schichten zu.
Erstelle eine neue Convolutional-Schicht, bei der in_channels dem out_channels der bestehenden Schicht entspricht, out_channels auf 32 gesetzt ist, stride und padding jeweils 1 sind und kernel_size 3 beträgt; weise sie conv2 zu.
Hänge die neue Schicht an das Modell an und nenne sie "conv2".

Interaktive praktische Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

# 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 bearbeiten und ausführen