Retour aux cas pratiques
PyTorchFondamental

Correction d’un modèle PyTorch sans mode évaluation activé

Scénario

Contexte : Un modèle affiche des résultats incohérents entre entraînement et test. Le Problème : Le modèle reste en mode entraînement pendant l’évaluation. Extrait logique actuelle : model.train() outputs = model(X_test) Contraintes : Utiliser model.eval() lors de l’inférence. Désactiver le calcul de gradients avec torch.no_grad(). Comparer performances avant/après correction. Livrable attendu : Corriger la phase d’évaluation pour garantir des résultats cohérents et reproductibles.

Compétences ciblées

evaluationmodel-modebugfixinferencetraining
    Correction d’un modèle PyTorch sans mode évaluation activé — mise en situation PyTorch | CodeDuel