Adăugarea unui nou strat convoluțional

Managerul de proiect ți-a pus la dispoziție un nou model CNN. Să analizăm arhitectura modelului și să îi adăugăm un nou strat convoluțional.

Modelul este disponibil ca CNNModel. Pachetele torch și torch.nn au fost importate ca nn.

Acest exercițiu face parte din cursul

Deep Learning pentru imagini cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

Instanțiază un model din clasa CNNModel și accesează straturile convoluționale.
Creează un nou strat convoluțional cu in_channels egal cu out_channels al stratului existent, out_channels setat la 32, stride și padding ambele setate la 1, și un kernel_size de 3; atribuie-l variabilei conv2.
Adaugă noul strat la model, numindu-l "conv2".

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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

Editează și rulează codul