Добавление нового свёрточного слоя

Руководитель проекта предоставил вам новую модель CNN. Давайте изучим архитектуру модели и добавим к ней новый свёрточный слой.

Модель доступна как CNNModel. Пакеты torch и torch.nn уже импортированы как nn.

Это упражнение является частью курса

Глубокое обучение для работы с изображениями на PyTorch

Посмотреть курс

Инструкции к упражнению

Создайте экземпляр модели из класса CNNModel и получите доступ к её свёрточным слоям.
Создайте новый свёрточный слой, в котором in_channels равен значению out_channels существующего слоя, out_channels равен 32, stride и padding равны 1, а kernel_size равен 3; присвойте его переменной conv2.
Добавьте новый слой в модель под именем "conv2".

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

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

Редактировать и запускать код