1. Nauka
    2. /
  2. Kursy
    3. /
  3. Głębokie uczenie z PyTorch – poziom średnio zaawansowany
Connected

ćwiczenie

Model z dwoma wejściami

Dane są gotowe – czas zbudować architekturę modelu z dwoma wejściami! W tym celu utworzysz klasę modelu z następującymi metodami:

  • .__init__(), w której zdefiniujesz podsieci, grupując warstwy; tutaj określasz dwie warstwy przetwarzające dwa wejścia oraz klasyfikator zwracający wynik klasyfikacji dla każdej klasy.

  • forward(), w której przepuścisz oba wejścia przez odpowiednie, wcześniej zdefiniowane podsieci, połączysz ich wyniki i przekażesz je do klasyfikatora.

torch.nn jest już zaimportowany jako nn. Do dzieła!

Instrukcje 1/3

undefined XP
    1
    2
    3
  • Zdefiniuj podsieci dla obrazu, alfabetu i klasyfikatora jako modele sekwencyjne, przypisując je odpowiednio do self.image_layer, self.alphabet_layer i self.classifier.