nn.Module

nn.Module est la classe de base de tout modèle PyTorch. On y déclare les couches dans __init__ et le calcul dans forward. Les paramètres sont enregistrés automatiquement, ce qui permet de les lister, de les transférer sur GPU et de sauvegarder l'état du modèle.

class Reseau(nn.Module):
    def __init__(self):
        super().__init__()
        self.fc = nn.Linear(10, 1)
    def forward(self, x): return self.fc(x)

Tout le glossaire