model.eval() en PyTorch

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.

Laisser un commentaire