Eine neue Faltungsschicht hinzufügen
Dein Projektleiter hat dir ein neues CNN-Modell gegeben. Schauen wir uns mal die Architektur des Modells an und fügen wir eine neue Faltungsschicht hinzu.
Das Modell ist als „ CNNModel “ verfügbar. Die Pakete „ torch “ und „ torch.nn “ sowie „ nn “ wurden importiert.
Diese Übung ist Teil des Kurses
Deep Learning für Bilder mit PyTorch
Anleitung zur Übung
- Erstelle ein Modell aus der Klasse „
CNNModel“ und greif auf die Faltungs-Layer zu. - Mach eine neue Faltungsschicht mit „
in_channels” gleich der „out_channels” der bestehenden Schicht, „out_channels” auf 32, „stride” und „padding” beide auf 1 und „kernel_size” auf 3; verknüpfe sie mit „conv2”. - Füge die neue Ebene zum Modell hinzu 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)