Aggiungere una nuova layer di convoluzione
Il project lead ti ha fornito un nuovo modello di CNN. Diamo un'occhiata all'architettura del modello e aggiungiamo una nuova layer di convoluzione.
Il modello è disponibile come CNNModel. I pacchetti torch e torch.nn come nn sono già stati importati.
Questo esercizio fa parte del corso
Deep Learning per Immagini con PyTorch
Istruzioni dell'esercizio
- Istanzia un modello dalla classe
CNNModele accedi alle layer di convoluzione. - Crea una nuova layer di convoluzione con
in_channelsuguale aout_channelsdella layer esistente,out_channelsimpostato a 32,strideepaddingentrambi uguali a 1 ekernel_sizepari a 3; assegnala aconv2. - Aggiungi la nuova layer al modello chiamandola
"conv2".
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# 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)