CommencezCommencez gratuitement

Initialisation des couches

L'initialisation des poids d'un réseau de neurones fait l'objet de recherches depuis de nombreuses années. Lors de l'entraînement d'un réseau, la méthode utilisée pour initialiser les poids a un impact direct sur la performance finale du réseau.

En tant que praticien en Machine Learning, vous devriez pouvoir expérimenter différentes stratégies d'initialisation. Dans cet exercice, vous créez un petit réseau de neurones composé de deux couches et vous décidez d'initialiser les poids de chaque couche avec la méthode uniforme.

Cette activité fait partie du cours

Introduction au Deep Learning avec PyTorch

Voir le cours

Instructions de l’exercice

  • Pour chaque couche (layer0 et layer1), utilisez la méthode d'initialisation uniforme pour initialiser les poids.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

layer0 = nn.Linear(16, 32)
layer1 = nn.Linear(32, 64)

# Use uniform initialization for layer0 and layer1 weights
nn.____.____(____)
nn.____.____(____)

model = nn.Sequential(layer0, layer1)
Modifier et exécuter le code