Ajouter une nouvelle couche de convolution

Votre responsable de projet vous a fourni un nouveau modèle de CNN. Examinons l'architecture du modèle et ajoutons-y une nouvelle couche de convolution.

Le modèle est disponible sous le nom CNNModel. Les modules torch et torch.nn (comme nn) ont été importés.

Cette activité fait partie du cours

Deep Learning pour les images avec PyTorch

Voir le cours

Instructions de l’exercice

Instanciez un modèle à partir de la classe CNNModel et accédez aux couches de convolution.
Créez une nouvelle couche de convolution avec in_channels égal au out_channels de la couche existante, out_channels défini à 32, stride et padding tous deux à 1, et un kernel_size de 3; assignez-la à conv2.
Ajoutez la nouvelle couche au modèle en la nommant "conv2".

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

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

Modifier et exécuter le code