Comment geler les paramètres de mon modèle PyTorch?
Solution :
Il arrive parfois que l’on veuille geler les paramètres d’un modèle pour ne pas les mettre à jour lorsque l’on entraîne le modèle.
Pour réaliser ce tour de passe-passe, il s’agit d’utiliser de mettre à False l’attribut requires_grad des paramètres du modèle.
Regarder l’exemple ci-dessous.
Exemple :
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Laisser un commentaire