Adicionando uma nova camada convolucional
O líder do seu projeto te deu um novo modelo CNN. Vamos dar uma olhada na arquitetura do modelo e adicionar uma nova camada convolucional a ele.
O modelo está disponível em CNNModel. Os pacotes torch e torch.nn, assim como nn, foram importados.
Este exercício faz parte do curso
Aprendizado profundo para imagens com PyTorch
Instruções do exercício
- Instancie um modelo da classe
CNNModele acesse as camadas convolucionais. - Crie uma nova camada convolucional com “
in_channels” igual à camada existente “out_channels”, “out_channels” definido como 32 e “stride” e “padding” ambos definidos como 1, e “kernel_size” como 3; atribua-a a “conv2”. - Adicione a nova camada ao modelo, chamando-a de “
"conv2"”.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# 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)