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
Instructions de l’exercice
- Pour chaque couche (
layer0etlayer1), 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)