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

Deep Learning für Bilder mit PyTorch

Kurs anzeigen

Anleitung zur Übung

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 Übung

Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.

# 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