1. Learn
    2. /
  2. Курси
    3. /
  3. Середній рівень Deep Learning з PyTorch
Connected

вправа

Модель із двома входами

Дані готові — час побудувати архітектуру моделі з двома входами! Для цього ви створите клас моделі з такими методами:

  • .__init__(), у якому ви визначите підмережі, згрупувавши шари; тут ви задасте два шари для обробки двох входів і класифікатор, що повертає оцінку належності до кожного класу.

  • forward(), у якому ви пропустите обидва входи через відповідні заздалегідь визначені підмережі, конкатенуєте їхні виходи та передасте результат до класифікатора.

torch.nn уже імпортовано для вас як nn. Рушаймо!

Інструкції 1/3

undefined XP
    1
    2
    3
  • Визначте підмережі для зображення, алфавіту та класифікатора як послідовні моделі й призначте їх до self.image_layer, self.alphabet_layer та self.classifier відповідно.