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

ćwiczenie

Tworzenie bloku sekwencyjnego

Postanowiłeś(-aś) przeprojektować szablon binarnego modelu CNN, tworząc blok warstw konwolucyjnych. Dzięki temu łatwiej ułożysz wiele warstw jedna po drugiej. Ulepszony model pozwoli ci swobodnie projektować różne architektury CNN.

torch oraz torch.nn jako nn zostały już zaimportowane.

Instrukcje

100 XP
  • W metodzie __init__() zdefiniuj blok warstw konwolucyjnych i przypisz go do self.conv_block.
  • W metodzie forward() przepuść dane wejściowe przez zdefiniowany blok konwolucyjny.