Додавання нового шару згортки

Ваш керівник проєкту надав вам нову модель 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)

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