1. Apprendre
    2. /
  2. Cours
    3. /
  3. Introduction au Deep Learning avec PyTorch
Connected

Exercice

Compter le nombre de paramètres

Les modèles de deep learning sont réputés pour avoir beaucoup de paramètres. Plus il y a de paramètres, plus la complexité de calcul augmente et plus l'entraînement est long. Il est donc essentiel de savoir combien de paramètres comporte votre modèle.

Dans cet exercice, vous allez d'abord calculer manuellement le nombre de paramètres. Ensuite, vous vérifierez votre résultat à l'aide de la méthode .numel().

Instructions 1/2

undefined XP
    1
    2

Question

Calculez manuellement le nombre de paramètres du modèle ci-dessous. Combien y en a-t-il? Utilisez la console comme calculatrice.

model = nn.Sequential(nn.Linear(9, 4),
                      nn.Linear(4, 2),
                      nn.Linear(2, 1))

Réponses possibles