Problème :
À quoi sert model.eval() en PyTorch?
Solution :
model.eval() sert à désactiver certaines couches du modèle model lorsqu’on n’est plus en train de l’entraîner.
En effet, certaines couches comme les couches de batch normalization ou de dropout sont souvent destinées uniquement à l’entraînement et peuvent générer des comportements non désirés lors de phases d’évaluation.
On utilisera donc cette ligne, en amont d’une phase d’évaluation, pour désactiver ce type de couches.
Compléments :

Laisser un commentaire