Initialisation des couches
L’initialisation des poids d’un réseau neuronal est au centre des préoccupations des chercheurs 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 du machine learning , vous devez être en mesure d'expérimenter plusieurs stratégies d'initialisation. Dans cet exercice, vous créez un petit réseau neuronal composé de deux couches et vous décidez d'initialiser les poids de chaque couche avec la méthode uniforme.
Cet exercice fait partie du cours
Introduction au deep learning avec PyTorch
Instructions
- Pour chaque couche (
layer0
etlayer1
), utilisez la méthode d'initialisation uniforme pour initialiser les poids.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
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)