Dodawanie nowej warstwy splotowej

Kierownik projektu udostępnił ci nowy model CNN. Przyjrzyjmy się architekturze tego modelu i dodajmy do niego nową warstwę splotową.

Model jest dostępny jako CNNModel. Pakiety torch oraz torch.nn jako nn zostały już zaimportowane.

To ćwiczenie jest częścią kursu

Głębokie uczenie dla obrazów z PyTorch

Zobacz kurs

Instrukcje do ćwiczenia

Utwórz instancję modelu z klasy CNNModel i uzyskaj dostęp do jego warstw splotowych.
Utwórz nową warstwę splotową, w której in_channels jest równe out_channels istniejącej warstwy, out_channels wynosi 32, a stride i padding mają wartość 1, oraz kernel_size równy 3; przypisz ją do zmiennej conv2.
Dołącz nową warstwę do modelu, nadając jej nazwę "conv2".

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

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

Edytuj i uruchom kod