1. Learn
    2. /
  2. Cursuri
    3. /
  3. Deep Learning intermediar cu PyTorch
Connected

exercițiu

Model cu două intrări

Cu datele pregătite, e momentul să construiești arhitectura modelului cu două intrări! Pentru aceasta, vei defini o clasă de model cu următoarele metode:

  • .__init__(), în care vei defini sub-rețelele prin gruparea straturilor; aici definești cele două straturi pentru procesarea celor două intrări și clasificatorul care returnează un scor de clasificare pentru fiecare clasă.

  • forward(), în care vei trece ambele intrări prin sub-rețelele predefinite corespunzătoare, vei concatena rezultatele și le vei transmite clasificatorului.

torch.nn este deja importat ca nn. Să începem!

Instrucțiuni 1/3

undefined XP
    1
    2
    3
  • Definește sub-rețelele pentru imagini, alfabet și clasificator ca modele secvențiale, atribuindu-le respectiv lui self.image_layer, self.alphabet_layer și self.classifier.