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

cvičení

Model se dvěma vstupy

Data jsou připravená, takže je čas sestavit architekturu modelu se dvěma vstupy! Vytvoříš třídu modelu s následujícími metodami:

  • .__init__(), ve které definuješ podsítě seskupením vrstev – tady nastavíš dvě vrstvy pro zpracování dvou vstupů a klasifikátor, který vrátí skóre klasifikace pro každou třídu.

  • forward(), ve které oba vstupy propustíš příslušnými podsítěmi, výstupy zřetězíš a předáš je klasifikátoru.

torch.nn je už naimportovaný jako nn. Jdeme na to!

Pokyny 1/3

undefined XP
    1
    2
    3
  • Definuj podsítě pro obrázky, abecedu a klasifikátor jako sekvenční modely a přiřaď je do self.image_layer, self.alphabet_layer a self.classifier.