ÎncepețiÎncepe gratuit

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!

Acest exercițiu face parte din cursul

Deep Learning intermediar cu PyTorch

Vezi cursul

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

class Net(nn.Module):
    def __init__(self):
        super().__init__()
        # Define sub-networks as sequential models
        ____ = ____(
            nn.Conv2d(1, 16, kernel_size=3, padding=1),
            nn.MaxPool2d(kernel_size=2),
            nn.ELU(),
            nn.Flatten(),
            nn.Linear(16*32*32, 128)
        )
        ____ = ____(
            nn.Linear(30, 8),
            nn.ELU(), 
        )
        ____ = ____(
            nn.Linear(128 + 8, 964), 
        )
Editează și rulează codul