Conception du modèle : la classe nn.Module

Construire un réseau de neurones dans PyTorch, c'est comme assembler des LEGO. Dans _init_, vous déclarez les pièces (couches) et en avant, vous spécifiez comment le tenseur d'entrée doit les parcourir pour devenir une prédiction. Le principal avantage de nn.Module est qu'il suit automatiquement tous les paramètres (poids) à l'intérieur de ses couches.

Full content is available with a subscription.
Get full access to all courses on the platform for one year with a single payment.
Unlike other platforms that charge per course, here you get everything for one price, and after one year of use there will be no automatic charge for the following year.