1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Intermediate Deep Learning with PyTorch
Connected

cvičení

Architektura modelu se dvěma výstupy

V tomto cvičení sestavíš architekturu vícevrstvé neuronové sítě s více výstupy, která bude schopná předpovídat znak i abecedu.

Připomeň si obecnou strukturu: v metodě .__init__() definuješ vrstvy, které se pak použijí ve forward passu. V metodě forward() nejprve projde vstupní obrázek několika vrstvami a získá se jeho embedding, který se poté předá dvěma samostatným klasifikačním vrstvám — jedné pro každý výstup.

torch.nn je již naimportován pod obvyklým aliasem, takže se pustíme do tvorby modelu!

Pokyny 1/2

undefined XP
    1
    2
  • Definuj self.classifier_alpha a self.classifier_char jako lineární vrstvy, jejichž vstupní velikost odpovídá výstupu vrstvy image_layer a výstupní velikost odpovídá počtu abeced (30), respektive počtu znaků (964).