Ajout d'une nouvelle couche convolutive
Votre chef de projet vous a fourni un nouveau modèle CNN. Examinons l'architecture du modèle et ajoutons-y une nouvelle couche convolutive.
Le modèle est disponible à l'adresse suivante : CNNModel. Les paquets torch et torch.nn ont été importés en tant que nn.
Cet exercice fait partie du cours
Deep learning pour les images avec PyTorch
Instructions
- Instancier un modèle à partir de la classe
CNNModelet accéder aux couches convolutives. - Créez une nouvelle couche convolutive avec l'
in_channelse égale à celle de la couche existante, l'out_channels, l'out_channels, l'strideet l'paddingtous deux définis sur 1, et l'kernel_sizesur 3 ; assignez-la à l'conv2. - Ajoutez le nouveau calque au modèle et nommez-le «
"conv2"».
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# 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)