Geler les Paramètres de mon Modèle PyTorch

Problème :

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 :

for params in model.parameters():
params.requires_grad = False
# Où model est un modèle que vous avez déjà pré-défini.

Laisser un commentaire