CommencerCommencer gratuitement

Rédaction de la boucle d'évaluation

Dans cet exercice, vous écrirez une boucle d’évaluation pour calculer la perte de validation. La boucle d’évaluation suit une structure similaire à celle de la boucle d’apprentissage, mais sans calcul de gradient ni mise à jour des poids.

model, validationloader, et le criterion de la fonction de perte ont déjà été définis pour gérer les prédictions, le chargement des données et le calcul des pertes.

Cet exercice fait partie du cours

Introduction au deep learning avec PyTorch

Afficher le cours

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Set the model to evaluation mode
____
validation_loss = 0.0

with torch.no_grad():
  
  for features, labels in validationloader:
    
      outputs = model(features)
      loss = criterion(outputs, labels)
      
      # Sum the current loss to the validation_loss variable
      validation_loss += ____
Modifier et exécuter le code