Añadir una nueva capa convolucional

Tu responsable de proyecto te ha proporcionado un nuevo modelo de CNN. Echa un vistazo a la arquitectura del modelo y añade una nueva capa convolucional.

El modelo está disponible como CNNModel. Los paquetes torch y torch.nn como nn ya se han importado.

Este ejercicio forma parte del curso

Deep Learning para imágenes con PyTorch

Ver curso

Instrucciones del ejercicio

Instancia un modelo de la clase CNNModel y accede a las capas convolucionales.
Crea una nueva capa convolucional con in_channels igual a los out_channels de la capa existente, out_channels establecido en 32, stride y padding ambos en 1, y kernel_size de 3; asígnala a conv2.
Añade la nueva capa al modelo, llamándola "conv2".

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

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

Editar y ejecutar código