1. Nauka
    2. /
  2. Kursy
    3. /
  3. Głębokie uczenie dla obrazów z PyTorch
Connected

ćwiczenie

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.

Instrukcje

100 XP
  • 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".