1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

cvičení

Vytvoření sekvenčního bloku

Rozhodl/a ses přepracovat šablonu svého binárního CNN modelu a vytvořit blok konvolučních vrstev. To ti umožní skládat více vrstev za sebou. S tímto vylepšeným modelem pak snadno navrhneš různé CNN architektury.

torch a torch.nn jako nn jsou již naimportovány.

Pokyny

100 XP
  • V metodě __init__() definuj blok konvolučních vrstev a přiřaď ho do self.conv_block.
  • Ve funkci forward() předej vstupy přes konvoluční blok, který jsi definoval/a.